Update app.py

app.py

@@ -6,12 +6,15 @@ from torchvision import transforms
 from transformers import AutoModelForImageClassification, AutoConfig
 import gradio as gr
 import spaces
+import os
+
+token = os.environ.get("HUGGINGFACE_HUB_TOKEN")
 
 
 
 model_id = "thelabel/240903-image-tagging"
-config = AutoConfig.from_pretrained(model_id)
-model = AutoModelForImageClassification.from_pretrained(model_id)
+config = AutoConfig.from_pretrained(model_id,token=token)
+model = AutoModelForImageClassification.from_pretrained(model_id,token=token)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)