Update app.py

app.py

@@ -1,29 +1,25 @@
 from flask import Flask, request, jsonify
-from transformers import pipeline
+from gradio_client import Client
 import os
 
-app = Flask(__name__)
-
-# Modeli başta yükle (sadece 1 kez)
-question_generator = pipeline("text2text-generation", model="memorease/flan5_memorease")
+app = Flask(_name_)
 
 @app.route("/ask", methods=["POST"])
 def ask_question():
     try:
+        client = Client("memorease/memorease-flan-t5")
         input_text = request.json.get("text")
         if not input_text:
             return jsonify({"error": "Missing 'text'"}), 400
-
-        result = question_generator(f"Generate a question from this sentence: {input_text}")
-        return jsonify({"question": result[0]['generated_text']})
+        result = client.predict(input_text, api_name="/predict")
+        return jsonify({"question": result})
     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
 
-if __name__ == "__main__":
+if _name_ == "_main_":
     port = int(os.environ.get("PORT", 7860))
-    app.run(host="0.0.0.0", port=port)
+    app.run(host="0.0.0.0", port=port)