api.py

from flask import Flask, request, jsonify
from app import generate_text

app = Flask(__name__)



@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)

app.py

@@ -0,0 +1,7 @@
+from transformers import pipeline
+
+model = pipeline("text-generation", model="gpt2")
+
+def generate_text(prompt, max_length=100):
+    output = model(prompt, max_length=max_length)
+    return output[0]["generated_text"]