原标题: 1. 导入OpenAI的GPT模型和tokenizer。
导读:
```pythonfrom transformers import GPT2LMHeadModel, GPT2Tokenizermodel_name = 'gpt2'model...
```python
from transformers import GPT2LMHeadModel, GPT2Tokenizer
model_name = 'gpt2'
model = GPT2LMHeadModel.from_pretrained(model_name)
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
```
2. 定义helper函数来生成回答。
def generate_response(input_text, max_length=50):
# 编码输入文本
input_ids = tokenizer.encode(input_text, return_tensors='pt')
# 生成回答
output = model.generate(input_ids, max_length=max_length,
num_return_sequences=1,
no_repeat_ngram_size=3,
early_stopping=True)
# 解码并返回回答文本
response_text = tokenizer.decode(output[0], skip_special_tokens=True)
return response_text
3. 提供用户交互界面,让用户输入问题并获取模型的回答。
print("ChatGPT体验程序")
while True:
user_input = input("你: ")
if user_input.lower() in ['exit', 'quit']:
print("ChatGPT程序结束。")
break
response = generate_response(user_input)
print("ChatGPT: " + response)
使用上述代码,在终端运行Python脚本,即可开始与ChatGPT进行对话,通过键盘输入问题后,ChatGPT会生成一个合适长度的回答,并显示在屏幕上,当想要退出对话时,可以输入"exit"或者"quit"命令。