from transformers import pipeline, set_seed
import torch
from flask import Flask, request, jsonify

# 加载GPT-2模型
generator = pipeline('text-generation', model='gpt2')

# 设置随机种子
set_seed(42)

# 创建Flask应用程序
app = Flask(__name__)

# 定义路由
@app.route('/chatbot', methods=['POST'])
def chatbot():
    # 获取请求中的文本
    text = request.json['text']
    
    # 使用GPT-2模型生成响应
    response = generator(text, max_length=50, do_sample=True, temperature=0.7)[0]['generated_text']
    
    # 返回响应
    return jsonify({'response': response})

# 启动应用程序
if __name__ == '__main__':
    app.run(host="0.0.0.0", port=7860, debug=True))