basic error handling

app.py

@@ -2,6 +2,7 @@
 import gradio as gr
 import requests
 from huggingface_hub import HfApi
+from huggingface_hub.errors import RepositoryNotFoundError
 import pandas as pd
 import plotly.express as px
 
@@ -52,8 +53,19 @@ def repo_size(r_type, r_id):
 
 
 def get_repo_info(r_type, r_id):
-    repo_sizes = repo_size(r_type, r_id)
-    repo_files_info = repo_files(r_type, r_id)
+    try:
+        repo_sizes = repo_size(r_type, r_id)
+        repo_files_info = repo_files(r_type, r_id)
+    except RepositoryNotFoundError:
+        gr.Warning(
+            "Repository not found. Make sure you've entered a valid repo ID and type that corresponds to the repository."
+        )
+        return (
+            gr.Row(visible=False),
+            gr.Dataframe(visible=False),
+            gr.Plot(visible=False),
+            gr.Dataframe(visible=False),
+        )
     rf_sizes_df = (
         pd.DataFrame(repo_files_info)
         .T.reset_index(names="ext")