python 对接chatgtp

要对接 ChatGPT，可以使用 OpenAI 的 API，或者使用 Python 的 transformers 库。以下是使用 transformers 库的示例代码：

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "microsoft/DialoGPT-small"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

def generate_response(user_input):
    input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
    chat_output = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
    return tokenizer.decode(chat_output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)

这个代码片段定义了一个函数 `generate_response`，输入用户的文本，使用 DialoGPT-small 模型生成一段回复。你可以使用其他的模型，只需要将 `model_name` 替换成相应的模型名称即可。