Update app.py

app.py

@@ -5,7 +5,7 @@ from huggingface_hub import list_models, list_datasets, list_spaces
 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: gr.OAuthToken):
+def get_total_storage(namespace, resource_type, oauth_token: Union[gr.OAuthToken, None]):
     if resource_type == "model":
         resources = list_models(author=namespace, token=oauth_token.token)
         url_base = "https://huggingface.co/api/models"
@@ -27,7 +27,7 @@ def get_total_storage(namespace, resource_type, oauth_token: gr.OAuthToken):
 
     return total_storage, len(resources)
 
-def get_report(namespace, oauth_token: Union[OAuthToken, None]):
+def get_report(namespace, oauth_token: Union[gr.OAuthToken, None]):
     # Fetch storage and counts for models, datasets, and spaces
     model_storage, n_models = get_total_storage(namespace, "model", oauth_token)
     dataset_storage, n_datasets = get_total_storage(namespace, "dataset", oauth_token)