python调用chatgtp

要使用Python调用ChatGPT，您需要安装Hugging Face的Transformers库。这个库提供了许多预训练的语言模型，其中包括ChatGPT。

下面是一个简单的Python脚本，可以使用ChatGPT来生成一些文本：

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

# 加载模型和分词器
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")

# 与用户交互并生成响应
while True:
    # 获取用户输入
    user_input = input("User: ")

    # 将用户输入编码为输入IDs
    input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors="pt")

    # 生成响应
    chat_response = model.generate(input_ids=input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)

    # 将响应解码为文本
    output = tokenizer.decode(chat_response[0], skip_special_tokens=True)

    # 打印响应
    print("Chatbot: " + output)

在这个脚本中，我们首先加载了ChatGPT的预训练模型和分词器。然后，我们使用一个简单的while循环，与用户交互并生成响应。在每次循环迭代中，我们首先获取用户输入，将其编码为输入IDs，然后使用模型生成响应。最后，我们将响应解码为文本，并打印出来。

请注意，这只是一个简单的示例，您可以根据您的需求进行更改和扩展。