Add error handling for model information retrieval in get_model_info function

utils.py

@@ -34,17 +34,21 @@ def load_json_results(file_path: Path, prepare_for_display=False, sort_col=None,
     return df
 
 def get_model_info(model_id, verbose=False):
-    model_info = api.model_info(model_id)
-    num_downloads = model_info.downloads
-    num_likes = model_info.likes
-    license = model_info.card_data["license"]
-    num_parameters = round(model_info.safetensors.total / 1e6)
-    supported_precisions = list(model_info.safetensors.parameters.keys())
-    if verbose:
-        print(f"Model '{model_id}' has {num_downloads} downloads, {num_likes} likes, and is licensed under {license}.")
-        print(f"The model has approximately {num_parameters:.2f} billion parameters.")
-        print(f"The model supports the following precisions: {supported_precisions}")
-    return num_downloads, num_likes, license, num_parameters, supported_precisions
+    try:
+        model_info = api.model_info(model_id)
+        num_downloads = model_info.downloads
+        num_likes = model_info.likes
+        license = model_info.card_data["license"]
+        num_parameters = round(model_info.safetensors.total / 1e6)
+        supported_precisions = list(model_info.safetensors.parameters.keys())
+        if verbose:
+            print(f"Model '{model_id}' has {num_downloads} downloads, {num_likes} likes, and is licensed under {license}.")
+            print(f"The model has approximately {num_parameters:.2f} billion parameters.")
+            print(f"The model supports the following precisions: {supported_precisions}")
+        return num_downloads, num_likes, license, num_parameters, supported_precisions
+    except Exception as e:
+        gr.Error(f"Error: Could not fetch model information. {str(e)}")
+        return 0, 0, "Unknown", 0, []
 
 def fetch_model_information(model_name):
     try: