将上面的代码的输入和输出改成直接输入输出
时间: 2023-07-16 14:17:10 浏览: 40
把代码改成直接输入输出的形式可以这样写:
```python
# 引入相应的库
from transformers import AutoTokenizer, AutoModelWithLMHead
# 加载预训练模型和分词器
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
model = AutoModelWithLMHead.from_pretrained("microsoft/DialoGPT-medium")
# 进行对话
print("你好,我是一个AI,我可以回答你的问题。如果你不知道该问什么,可以先和我聊聊天哦!")
while True:
# 用户输入
user_input = input("用户:")
# 将用户输入转换成模型可接受的格式
input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
# 生成回复
bot_output = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
# 将回复转换成可读的格式
bot_reply = tokenizer.decode(bot_output[0], skip_special_tokens=True)
# 输出回复
print("AI:" + bot_reply)
```
这样代码就能够直接从命令行获取用户输入,并输出对话结果了。