Update app.py

app.py

@@ -1,21 +1,25 @@
 from flask import Flask, request, jsonify
-from gradio_client import Client
+from transformers import pipeline
 import os
 
 app = Flask(__name__)
 
+# Modeli başta yükle (sadece 1 kez)
+question_generator = pipeline("text2text-generation", model="memorease/flan5_memorease")
+
 @app.route("/ask", methods=["POST"])
 def ask_question():
     try:
-        client = Client("memorease/flan5_memorease")
         input_text = request.json.get("text")
         if not input_text:
             return jsonify({"error": "Missing 'text'"}), 400
-        result = client.predict(input_text, api_name="/predict")
-        return jsonify({"question": result})
+
+        result = question_generator(f"Generate a question from this sentence: {input_text}")
+        return jsonify({"question": result[0]['generated_text']})
     except Exception as e:
+        print("SERVER ERROR:", str(e))
         return jsonify({"error": str(e)}), 500
-        
+
 @app.route("/", methods=["GET"])
 def index():
     return "Flan5 Memorease Docker Space is running!", 200