Update app.py

app.py

@@ -1,24 +1,21 @@
 from flask import Flask, request, jsonify
-from transformers import pipeline
+from gradio_client import Client
 import os
 
 app = Flask(__name__)
 
-pipe = pipeline("text2text-generation", model="memorease/flan5_memorease", token=os.environ.get("HF_TOKEN"))
-
 @app.route("/ask", methods=["POST"])
 def ask_question():
     try:
+        client = Client("memorease/flan5_memorease")
         input_text = request.json.get("text")
-        result = pipe(f"Generate a question about: {input_text}")[0]["generated_text"]
+        if not input_text:
+            return jsonify({"error": "Missing 'text'"}), 400
+        result = client.predict(input_text, api_name="/predict")
         return jsonify({"question": result})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-@app.route("/", methods=["GET"])
-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)