Update app.py

app.py

@@ -48,9 +48,12 @@ def initialize_agent(endpoint_uri: str):
     ) 
     
 
-def wake_up_endpoint(endpoint_uri, ui, max_wait=30):
+def wake_up_endpoint(endpoint_uri, ui, max_wait=300):
     """Poll the endpoint until it responds OK or timeout."""
 
+    
+    # Enforce minimum wait time of 60 seconds
+    max_wait = max(max_wait, 60)
     # Get token from environment variable or secrets store
     hf_token = os.environ.get("HF_TOKEN")  # Or replace with your secret loading logic
     headers = {"Authorization": f"Bearer {hf_token}"} if hf_token else {}