Update app.py

app.py

@@ -6,14 +6,15 @@ from typing import Union
 
 # Helper function to get the total storage for models, datasets, or spaces
 def get_total_storage(namespace, resource_type, oauth_token: Union[gr.OAuthToken, None]):
+    token = oauth_token.token if oauth_token else None
     if resource_type == "model":
-        resources = list(list_models(author=namespace, token=oauth_token.token))
+        resources = list(list_models(author=namespace, token=token))
         url_base = "https://huggingface.co/api/models"
     elif resource_type == "dataset":
-        resources = list(list_datasets(author=namespace, token=oauth_token.token))
+        resources = list(list_datasets(author=namespace, token=token))
         url_base = "https://huggingface.co/api/datasets"
     elif resource_type == "space":
-        resources = list(list_spaces(author=namespace, token=oauth_token.token))
+        resources = list(list_spaces(author=namespace, token=token))
         url_base = "https://huggingface.co/api/spaces"
     
     total_storage = 0