能不能把上面的代码改成自己输入的形式
时间: 2024-02-09 21:11:31 浏览: 21
当然可以,你可以在代码中使用 input() 函数来获取用户输入的内容。例如,你可以使用以下代码来获取用户输入的文本,并将其作为对话的起始点:
```python
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
# 获取用户输入的文本
user_text = input("请输入您想说的话:")
# 将用户输入的文本编码成对应的 token ID
input_ids = tokenizer.encode(user_text + tokenizer.eos_token, return_tensors='pt')
# 生成 AI 的回复
bot_text = model.generate(input_ids=input_ids, max_length=1024, pad_token_id=tokenizer.eos_token_id)
# 将 AI 回复解码成文本并输出
print(tokenizer.decode(bot_text.squeeze(), skip_special_tokens=True))
```
当用户运行上述代码时,程序会提示用户输入一段话。用户输入的内容将被传递给 AI 模型进行处理,并生成回复。最后,程序将 AI 的回复输出到屏幕上。