save2

app.py

@@ -1,6 +1,20 @@
 import torch
 from transformers import AutoProcessor, AutoModelForVision2Seq
+from huggingface_hub import login
 import gradio as gr
+import os
+
+# ----------------------------
+# AUTHENTICATION
+# ----------------------------
+# Option 1: Use HF token from environment variable (recommended for Spaces)
+HF_TOKEN = os.getenv("HF_TOKEN")
+if HF_TOKEN:
+    login(token=HF_TOKEN)
+else:
+    # Option 2: Interactive login (for local testing)
+    print("No HF_TOKEN found. Please log in manually.")
+    login()
 
 # ----------------------------
 # CONFIG
@@ -17,8 +31,12 @@ model = AutoModelForVision2Seq.from_pretrained(
     MODEL_NAME,
     torch_dtype=DTYPE,
     device_map="auto",
+    token=HF_TOKEN,  # Pass token explicitly
+)
+processor = AutoProcessor.from_pretrained(
+    MODEL_NAME,
+    token=HF_TOKEN,
 )
-processor = AutoProcessor.from_pretrained(MODEL_NAME)
 print("Model loaded successfully.")
 
 # ----------------------------
@@ -68,4 +86,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()