Improving verbosity. Adding checks for missing URL.

app.py

@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
 
 verbose = True
 
-if verbose: print("Running app.py now\n")
+if verbose: print("Running app.py.\n")
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -82,7 +82,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-if verbose: print("Defining MODEL_IDS now\n")
+if verbose: print("Defining MODEL_IDS.\n")
 
 MODEL_IDS = [
     'https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud/',
@@ -99,6 +99,8 @@ def is_model_overloaded(model_url):
         response = requests.post(model_url, json={"inputs": "Test"})
         if response.status_code == 503:  # 503 Service Unavailable = Overloaded
             return True
+        if response.status_code == 404:  # 404 Client Error: Not Found 
+            return True
         return False
     except requests.RequestException:
         return True  # if there are an error is overloaded
@@ -106,12 +108,12 @@ def is_model_overloaded(model_url):
 def get_available_model():
     """Select the first model available from the list."""
     for model_url in MODEL_IDS:
-        print("trying",model_url,"\n")
+        print("  trying",model_url,"\n")
         if not is_model_overloaded(model_url):
             return model_url
     return MODEL_IDS[0]  # if all are failing, use the first model by dfault
 
-if verbose: print("Checking available models now\n")
+if verbose: print("Checking available models.\n")
 
 selected_model_id = get_available_model()