Update app.py

app.py

@@ -2,11 +2,20 @@ import spaces
 from snac import SNAC
 import torch
 import gradio as gr
+import os
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import snapshot_download
 from dotenv import load_dotenv
 load_dotenv()
 
+# Check if HF_TOKEN is available
+token = os.getenv("HF_TOKEN")
+if token:
+    from huggingface_hub import login
+    login(token=token)
+else:
+    print("⚠️  No HF_TOKEN found – gated model will fail.")
+
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"