Create app.py

app.py

@@ -0,0 +1,28 @@
+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(debug=True)