Update model/model_utils.py

model/model_utils.py

@@ -1,10 +1,13 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import os
 
 def load_model():
     model_name = "bigcode/starcoder"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
+    hf_token = os.getenv("HF_TOKEN")  # ✅ Securely pulled from Space Secrets
+
+    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token, trust_remote_code=True)
     model.eval()
 
     device = "cuda" if torch.cuda.is_available() else "cpu"