Update app.py

app.py

@@ -1,32 +1,26 @@
 from flask import Flask, request, jsonify
 from gradio_client import Client
 import os
-from transformers import pipeline
 
 app = Flask(__name__)
 
-question_generator = pipeline("text2text-generation", model="memorease/memorease-flan-t5")
-
 @app.route("/ask", methods=["POST"])
 def ask_question():
     try:
+        client = Client("memorease/memorease-flan-t5")
         input_text = request.json.get("text")
-        print("INPUT:", input_text)
-
         if not input_text:
             return jsonify({"error": "Missing 'text'"}), 400
-
-        result = question_generator(f"Generate a question from this sentence: {input_text}")
-        print("RESULT:", result)
-        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
+def root_check():
+    return jsonify({"status": "running"})
+
 
 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)