Create app.py

app.py

@@ -1,7 +1,22 @@
-from transformers import pipeline
+from flask import Flask, request, jsonify
+from app import generate_text
 
-model = pipeline("text-generation", model="gpt2")
+app = Flask(__name__)
 
-def generate_text(prompt, max_length=100):
-    output = model(prompt, max_length=max_length)
-    return output[0]["generated_text"]
+@app.route("/generate", methods=["POST"])
+def generate():
+    data = request.json
+    prompt = data.get("prompt", "")
+    max_length = data.get("max_length", 100)
+    
+    if not prompt:
+        return jsonify({"error": "Prompt is required"}), 400
+    
+    try:
+        result = generate_text(prompt, max_length)
+        return jsonify({"generated_text": result})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8080)