Update app.py

app.py

@@ -47,6 +47,11 @@ def initialize_agent(endpoint_uri: str):
 
 def wake_up_endpoint(endpoint_uri, max_wait=30):
     """Poll the endpoint until it responds OK or timeout."""
+
+    # 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 {}
+    
     payload = {"inputs": "ping"}  # Adjust to minimal valid input
     start = time.time()