Update app.py

app.py

@@ -16,8 +16,20 @@ import spaces
 model_id = "gv-hf/paligemma2-10b-mix-448"
 COLORS = ['#4285f4', '#db4437', '#f4b400', '#0f9d58', '#e48ef1']
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)
-processor = PaliGemmaProcessor.from_pretrained(model_id)
+HF_KEY = os.getenv("HF_KEY")
+if not HF_KEY:
+    raise ValueError("Please set the HF_KEY environment variable with your Hugging Face API token")
+
+model = PaliGemmaForConditionalGeneration.from_pretrained(
+    model_id,
+    token=HF_KEY,
+    trust_remote_code=True
+).eval().to(device)
+
+processor = PaliGemmaProcessor.from_pretrained(
+    model_id,
+    token=HF_KEY,
+    trust_remote_code=True)
 
 @spaces.GPU
 def infer(image: PIL.Image.Image, text: str, max_new_tokens: int) -> str: