Update app.py

app.py

@@ -4,18 +4,32 @@ import os
 
 app = Flask(__name__)
 
+HF_TOKEN = os.environ.get("HF_TOKEN")
+client = None  # global cache
+
 @app.route("/ask", methods=["POST"])
 def ask_question():
+    global client
     try:
-        client = Client("memorease/flan5_memorease")
+        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")
         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)
+    app.run(host="0.0.0.0", port=port)