Update app.py

app.py

@@ -22,11 +22,11 @@ from Gradio_UI import GradioUI
 endpoint_uri = load_huggingface_model()
 
 # Test the endpoint
-response = check_public_endpoint(endpoint_uri)
 
-if response.status is False:
-     create_failed_gradio_ui()
+status_info = check_public_endpoint(endpoint_uri)
 
+if status_info["status"] is False:
+     create_failed_gradio_ui()
 else: 
     
     model = HfApiModel(
@@ -54,6 +54,6 @@ else:
      
     )
     
-    
+    # Launch the UI earlier or without passing through the Agent
     GradioUI(agent).launch()