python 调用chatgpt

要调用ChatGPT，需要先安装相应的库，如transformers和torch。然后可以使用以下代码进行调用：

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
# 输入对话历史和问题
input_str = "Hello, how are you?"
input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors='pt')

# 生成回答
sample_output = model.generate(
    input_ids,
    do_sample=True, 
    max_length=100, 
    top_p=0.92, 
    top_k=50
)

# 解码回答
output_str = tokenizer.decode(sample_output[0], skip_special_tokens=True)
print(output_str)

这里使用了Microsoft开发的DialoGPT-medium模型，可以根据需要选择其他模型。