当前位置:   article > 正文

只需5分钟,带你快速创建AI聊天机器人_gradio.chatbot

gradio.chatbot

本文将介绍使用Hugging Face和Gradio在5分钟内建立AI聊天机器人,通过这个简单易懂的教程,学习直接在浏览器中使用低代码技术创建Gradio聊天机器人。

微信搜索关注《Python学研大本营》,加入读者群,分享更多精彩

图片

本简短教程将使用Microsoft DialoGPT模型、Hugging Face Space和Gradio接口建立一个简单的聊天机器人。使用类似的技术,你将可以在5分钟内开发和自定义自己的应用程序。

图片

1.创建新空间

  1. 前往hf.co并创建一个免费账户。然后,点击右上角的显示图像并选择“New Space”选项。

  2. 在表格中填写应用程序名称、许可证、空间硬件和可见性。

图片

图片来自Space

  1. 按“Create Space”以初始化该应用程序。

  2. 用户可以从本地系统克隆存储库并推送文件,也可以使用浏览器在Hugging Face上创建和编辑文件。

图片

图片来自AI ChatBot

2.创建ChatBot应用程序文件

我们将点击“Files”选项卡 > “+ Add file” > “Create a new file”。

图片

图片来自kingabzpro/AI-ChatBot

创建Gradio接口。你可以复制我的代码。

【Gradio】:https://www.gradio.app/

图片

图片来自app.py

现在已经加载了“microsoft/DialoGPT-large”分词器和模型,并创建了一个predict函数来获取响应并创建历史记录。

  1. from transformers import AutoModelForCausalLM, AutoTokenizer
  2. import gradio as gr
  3. import torch
  4. title = "
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/374587
    推荐阅读
    相关标签