Update app.py

app.py

@@ -23,7 +23,9 @@ def query(payload):
 	response = requests.post(endpoint_url, headers=headers, json=payload)
 	return response.json()
 
-endpoint = get_inference_endpoint(endpoint_name, token=personal_secret_token)
+def get_status():
+    endpoint = get_inference_endpoint(endpoint_name, token=personal_secret_token)
+    return endpoint.status
 
 def respond(
     message,
@@ -36,13 +38,13 @@ def respond(
 ):
     progress(0, desc="Starting")
 
-    if endpoint.status != "running":
+    if get_status() != "running":
         endpoint.client.post(json={
             "inputs": "wake up!"
         })
         progress(0.25, desc="Waking up model")
         
-    while endpoint.status != "running":
+    while get_status() != "running":
         time.sleep(1)
 
     progress(0.5, desc="Generating")