handling exception in list repo refs more gracefully

app.py

@@ -35,7 +35,11 @@ def repo_files(r_type: str, r_id: str) -> dict:
 
 
 def repo_size(r_type, r_id):
-    r_refs = HF_API.list_repo_refs(repo_id=r_id, repo_type=r_type)
+    try:
+        r_refs = HF_API.list_repo_refs(repo_id=r_id, repo_type=r_type)
+    except RepositoryNotFoundError:
+        gr.Warning(f"Repository is gated, branch information for {r_id} not available.")
+        return {}
     repo_sizes = {}
     for branch in r_refs.branches:
         try:
@@ -44,10 +48,11 @@ def repo_size(r_type, r_id):
                 timeout=1000,
             )
             response = response.json()
-            # get the status code
         except Exception:
             response = {}
-        if response.get("error") and "restricted" in response.get("error"):
+        if response.get("error") and (
+            "restricted" in response.get("error") or "gated" in response.get("error")
+        ):
             gr.Warning(f"Branch information for {r_id} not available.")
             return {}
         size = response.get("size")