能给我一个稍微靠谱点的代码吗
时间: 2023-06-15 07:06:29 浏览: 110
参考的代码
这里是一个使用Hugging Face Transformers库的Python代码示例,可以让您与预训练的GPT-2模型进行交互式会话:
```python
from transformers import pipeline, set_seed
# 设置随机种子,以确保生成的答案一致
set_seed(42)
# 加载预训练的GPT-2模型和tokenizer
generator = pipeline('text-generation', model='gpt2')
# 开始对话
conversation_history = []
while True:
# 获取用户输入
user_input = input("You: ")
# 添加用户输入到对话历史
conversation_history.append(user_input)
# 生成GPT-2模型的回复
bot_response = generator(conversation_history)[-1]['generated_text']
# 打印回复并将其添加到对话历史
print("Bot:", bot_response.strip())
conversation_history.append(bot_response)
```
这个代码示例使用了Hugging Face Transformers库中的`pipeline`函数来加载预训练的GPT-2模型和相应的tokenizer。然后,它使用一个while循环来启动一个交互式会话,其中用户输入会被添加到对话历史中,然后GPT-2模型会根据整个对话历史来生成一个回复。生成的回复会被打印出来,并添加到对话历史中,以便下一轮交互。
阅读全文