Update app.py

app.py

@@ -12,13 +12,25 @@ if not HF_TOKEN:
     st.error("❌ Hugging Face API key not found! Set it as 'HF_KEY' in Spaces secrets.")
     st.stop()
 
-# Load the model and processor
+# Load the model and processor with authentication
 @st.cache_resource
 def load_model():
     model_id = "google/paligemma2-3b-mix-224"
-    model = PaliGemmaForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto").eval()
-    processor = PaliGemmaProcessor.from_pretrained(model_id)
-    return processor, model
+    
+    try:
+        model = PaliGemmaForConditionalGeneration.from_pretrained(
+            model_id, token=HF_TOKEN, torch_dtype=torch.bfloat16, device_map="auto"
+        ).eval()
+
+        processor = PaliGemmaProcessor.from_pretrained(
+            model_id, token=HF_TOKEN
+        )
+
+        return processor, model
+
+    except Exception as e:
+        st.error(f"❌ Error loading model: {str(e)}")
+        st.stop()
 
 processor, model = load_model()