教你5分鐘內(nèi)使用Hugging Face和 Gradio 構(gòu)建 AI 聊天機(jī)器人

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

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

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

2. 在表單中填寫應(yīng)用程序名稱、許可證、空間硬件和可見性。

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

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

圖片來(lái)自AI ChatBot

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

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

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

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


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