Update status_check.py

status_check.py

@@ -101,18 +101,20 @@ def run_status_checks():
 
     # Wake it up before health check
     wake_up_successful = wake_up_endpoint(endpoint_uri, ui)
-    
-    if not wake_up_successful:
-        ui.append("Warning: Could not wake up the endpoint. Continuing.")
+    success, error_msg = wake_up_endpoint(endpoint_uri, ui)
+i 
+    if not success:
+        ui.append("Warning: Could not wake up the endpoint. Exiting.")
+        
     else:     
         ui.append("✅ End point responded OK.")
     
-    is_healthy, status_info = check_model_endpoint(endpoint_uri) # Test the endpoint
-    
-    if not is_healthy:
-        from checks.failed_check import create_failed_gradio_ui
-        interface = create_failed_gradio_ui(status_info)
-        interface.launch(show_error=True, share=True)
-        return None
-    
-    return endpoint_uri
+        is_healthy, status_info = check_model_endpoint(endpoint_uri) # Test the endpoint
+        
+        if not is_healthy:
+            from checks.failed_check import create_failed_gradio_ui
+            interface = create_failed_gradio_ui(status_info)
+            interface.launch(show_error=True, share=True)
+            return None
+        
+        return endpoint_uri