Handle secrets

app.py

@@ -4,15 +4,20 @@ from PIL import Image
 from transformers import ViTForImageClassification, ViTImageProcessor
 import requests
 import json
+import os
+
+hf_token = os.getenv("HF_TOKEN")
+
 
 # Load model and processor from Hugging Face Hub
-model = ViTForImageClassification.from_pretrained("patcdaniel/phytoViT_508k_20250611")
-processor = ViTImageProcessor.from_pretrained("patcdaniel/phytoViT_508k_20250611")
+model = ViTForImageClassification.from_pretrained("patcdaniel/phytoViT_508k_20250611",token=hf_token)
+processor = ViTImageProcessor.from_pretrained("patcdaniel/phytoViT_508k_20250611",token=hf_token)
 model.eval()
 
 # Load class labels from hosted file
 LABELS_URL = "https://huggingface.co/patcdaniel/phytoViT_508k_20250611/resolve/main/label_names.json"
-class_labels = requests.get(LABELS_URL).json()
+headers = {"Authorization": f"Bearer {hf_token}"}
+class_labels = requests.get(LABELS_URL, headers=headers).json()
 
 def predict(image):
     image = image.convert("RGB")