Update app.py

app.py

@@ -1,30 +1,20 @@
 from flask import Flask, request, jsonify
-from gradio_client import Client
+from transformers import pipeline
 import os
 
 app = Flask(__name__)
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
-client = None  # global cache
+pipe = pipeline("text2text-generation", model="memorease/flan5_memorease", token=os.environ.get("HF_TOKEN"))
 
 @app.route("/ask", methods=["POST"])
 def ask_question():
-    global client
     try:
-        if client is None:
-            client = Client("memorease/flan5_memorease", hf_token=HF_TOKEN)  # lazy init
-
         input_text = request.json.get("text")
-        if not input_text:
-            return jsonify({"error": "Missing 'text'"}), 400
-
-        result = client.predict(input_text, api_name="/predict")
+        result = pipe(f"Generate a question about: {input_text}")[0]["generated_text"]
         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"})