教你5分鐘內使用Hugging Face和 Gradio 構建 AI 聊天機器人

這個(gè)簡(jiǎn)短的教程將使用Microsoft DialoGPT模型，擁抱面部空間和Gradio干擾構建一個(gè)簡(jiǎn)單的聊天機器人。您將能夠使用類(lèi)似的技術(shù)在 5 分鐘內開(kāi)發(fā)和自定義您自己的應用程序。

推薦：使用NSDT場(chǎng)景編輯器助你快速搭建可二次編輯的3D應用場(chǎng)景

1. 轉到 hf.co 并創(chuàng )建一個(gè)免費帳戶(hù)。之后，單擊右上角的顯示圖像，然后選擇“新空間”選項。

2. 在表單中填寫(xiě)應用程序名稱(chēng)、許可證、空間硬件和可見(jiàn)性。

1. 按“創(chuàng )建空間”以初始化應用程序。

2. 您可以克隆存儲庫并從本地系統推送文件，或者使用瀏覽器在擁抱臉上創(chuàng )建和編輯文件。

圖片來(lái)自AI ChatBot

我們將單擊“文件”選項卡>+添加文件>創(chuàng )建新文件。

創(chuàng )建 Gradio 接口。你可以復制我的代碼。

我已經(jīng)加載了“microsoft/dialopGPT-large”標記器和模型，并創(chuàng )建了一個(gè)“預測”函數來(lái)獲取響應和創(chuàng )建歷史記錄。

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


title = "
				
*博客內容為網(wǎng)友個(gè)人發(fā)布，僅代表博主個(gè)人觀(guān)點(diǎn)，如有侵權請聯(lián)系工作人員刪除。