Fix authentication for private model

- Add HF_TOKEN environment variable support
- Fix deprecated torch_dtype to dtype
- Add token parameter to model and processor loading

app.py

@@ -13,17 +13,28 @@ import os
 # Model configuration
 MODEL_ID = "AKSazgar/ECG-Instruct-Llama-3.2-11B-Vision"
 
+# Get HuggingFace token from environment variable
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+
+if not HF_TOKEN:
+    raise ValueError(
+        "HF_TOKEN environment variable is not set. "
+        "Please add your HuggingFace token to the Space secrets. "
+        "Go to Settings > Variables and secrets > Add a secret"
+    )
+
 print(f"Loading model: {MODEL_ID}")
 print("This may take a few minutes on first load...")
 
-# Load model and processor
+# Load model and processor with authentication token
 model = MllamaForConditionalGeneration.from_pretrained(
     MODEL_ID,
-    torch_dtype=torch.bfloat16,
+    dtype=torch.bfloat16,
     device_map="auto",
+    token=HF_TOKEN,
 )
 
-processor = AutoProcessor.from_pretrained(MODEL_ID)
+processor = AutoProcessor.from_pretrained(MODEL_ID, token=HF_TOKEN)
 
 print("Model loaded successfully!")