Update app.py

app.py

@@ -1,15 +1,11 @@
 from flask import Flask, request, jsonify
 from gradio_client import Client
 import os
+from transformers import pipeline
 
 app = Flask(__name__)
 
-try:
-    print(">>> INITIALIZING CLIENT...")
-    client = Client("memorease/memorease-flan-t5")
-except Exception as e:
-    print(">>> CLIENT INIT ERROR:", str(e))
-    client = None  # fallback
+question_generator = pipeline("text2text-generation", model="memorease/memorease-flan-t5")
 
 @app.route("/ask", methods=["POST"])
 def ask_question():
@@ -20,13 +16,9 @@ def ask_question():
         if not input_text:
             return jsonify({"error": "Missing 'text'"}), 400
 
-        if client is None:
-            return jsonify({"error": "Model client failed to initialize."}), 500
-
-        result = client.predict(input_text, api_name="/predict")
-
+        result = question_generator(f"Generate a question from this sentence: {input_text}")
         print("RESULT:", result)
-        return jsonify({"question": result})
+        return jsonify({"question": result[0]["generated_text"]})
     except Exception as e:
         print("SERVER ERROR:", str(e))
         return jsonify({"error": str(e)}), 500