Update app.py

app.py

@@ -7,7 +7,7 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-verbose = True
+global verbose = True
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -51,23 +51,24 @@ def is_model_overloaded(model_url):
     """Verify if the model is overloaded doing a test call."""
     try:
         response = requests.post(model_url, json={"inputs": "Test"})
-        if verbose: 
-            print(response.status_code)
-        if response.status_code == 503:  # 503 Service Unavailable = Overloaded
-            return True
-        if response.status_code == 404:  # 404 Client Error: Not Found 
-            return True
-        if response.status_code == 424:  # 424 Client Error: Failed Dependency for url:
+        if verbose:
+            print(f"URL: {model_url}")
+            print(f"Response Status Code: {response.status_code}")
+        
+        if response.status_code in [503, 404, 424]:
             return True
         return False
-    except requests.RequestException:
-        return True  # if there are an error is overloaded
+    except requests.RequestException as e:
+        if verbose:
+            print(f"Error accessing {model_url}: {e}")
+        return True  # If there is an error, consider it overloaded
+
 
 def get_available_model():
     """Select the first model available from the list."""
     for model_url in MODEL_IDS:
         print("trying",model_url)
-        if not is_model_overloaded(model_url):
+        if not is_model_overloaded(model_url)):
             return model_url
     return MODEL_IDS[0]  # if all are failing, use the first model by default
 
@@ -75,6 +76,8 @@ if verbose: print("Checking available models.")
 
 selected_model_id = get_available_model()
 
+if verbose: print("Using model {selected_model_id}")
+    
 model = HfApiModel(
     max_tokens=1048,
     temperature=0.5,