raise exception for invalid token

app.py

@@ -10,7 +10,7 @@ import pandas as pd
 import os
 import backoff
 from functools import lru_cache
-from huggingface_hub import list_models, ModelFilter
+from huggingface_hub import list_models, ModelFilter, login
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 
@@ -78,7 +78,10 @@ def predict_subset(model_id, token):
         raise gr.Error(
             f"model_id {model_id} is not a valid image classification model id"
         )
-
+    try:
+        login(token)
+    except ValueError:
+        raise gr.Error("Invalid Hub token")
     API_URL = f"https://api-inference.huggingface.co/models/{model_id}"
     headers = {"Authorization": f"Bearer {token}"}