基于chatgpt的聊天机器人

基于chatgpt的聊天机器人

以下是一个基于Python的示例代码，展示如何在本地环境中使用预训练的ChatGPT模型构建一个简单的聊天机器人应用程序。在这个示例中，我们将使用Hugging Face Transformers库加载GPT-2模型，并创建一个基于命令行的用户界面。

步骤 1: 安装必要的库
确保您已经安装了以下库：

pip install transformers
pip install torch
步骤 2: 编写代码
from transformers import GPT2LMHeadModel, GPT2Tokenizer

加载预训练的GPT-2模型和分词器

model = GPT2LMHeadModel.from_pretrained(“gpt2”)
tokenizer = GPT2Tokenizer.from_pretrained(“gpt2”)

设置模型为eval模式

model.eval()

对话函数

def chatbot():
    user_input = input("You: ")
    input_ids = tokenizer.encode("User: " + user_input + " ChatGPT:")
    input_ids = torch.tensor(input_ids