Spaces:
Runtime error
Runtime error
| 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__) | |
| # 定义路由 | |
| 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)) | |