新闻  |   论坛  |   博客  |   在线研讨会
教你5分钟内使用Hugging Face和 Gradio 构建 AI 聊天机器人
ygtu | 2023-08-04 17:38:11    阅读:344   发布文章

这个简短的教程将使用Microsoft DialoGPT模型,拥抱面部空间和Gradio干扰构建一个简单的聊天机器人。您将能够使用类似的技术在 5 分钟内开发和自定义您自己的应用程序。

教你5分钟内使用Hugging Face和 Gradio 构建 AI 聊天机器人

推荐:使用NSDT场景编辑器助你快速搭建可二次编辑的3D应用场景

1. 创建新空间
  1. 转到 hf.co 并创建一个免费帐户。之后,单击右上角的显示图像,然后选择“新空间”选项。

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


Build AI Chatbot in 5 Minutes with Hugging Face and Gradio
  1. 按“创建空间”以初始化应用程序。

  2. 您可以克隆存储库并从本地系统推送文件,或者使用浏览器在拥抱脸上创建和编辑文件。

Build AI Chatbot in 5 Minutes with Hugging Face and Gradio


图片来自AI ChatBot

2.创建聊天机器人应用程序文件

我们将单击“文件”选项卡>+添加文件>创建新文件。

Build AI Chatbot in 5 Minutes with Hugging Face and Gradio

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

Build AI Chatbot in 5 Minutes with Hugging Face and Gradio

我已经加载了“microsoft/dialopGPT-large”标记器和模型,并创建了一个“预测”函数来获取响应和创建历史记录。

from transformers import AutoModelForCausalLM, AutoTokenizer
import gradio as gr
import torch


title = "        	

*博客内容为网友个人发布,仅代表博主个人观点,如有侵权请联系工作人员删除。

参与讨论
登录后参与讨论
推荐文章
最近访客