Update app.py

app.py

@@ -7,15 +7,24 @@ import spaces  # Mandatory library for Hugging Face ZeroGPU [1]
 
 MODEL_ID = "DevStudio-AI/Devstudio-Coder-1.5B"
 
+# Fetch your secure token from the Space Secrets environment
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
 print("Loading tokenizer and base model...")
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+
+# Pass the token parameter to ensure complete, authenticated downloads
+tokenizer = AutoTokenizer.from_pretrained(
+    MODEL_ID, 
+    token=HF_TOKEN
+)
 
 # We load the model in 16-bit on CPU first
 # ZeroGPU will automatically move the model to the GPU when the decorated function runs [1]
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
     torch_dtype=torch.float16,
-    device_map="cpu"
+    device_map="cpu",
+    token=HF_TOKEN
 )
 print("Model successfully loaded on CPU. Awaiting ZeroGPU allocation...")