Update model/model_utils.py

model/model_utils.py

@@ -4,10 +4,10 @@ import os
 
 def load_model():
     model_name = "bigcode/starcoder"
-    hf_token = os.getenv("HF_TOKEN")  # Secure access to gated model
+    hf_token = os.getenv("HF_TOKEN")
 
-    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)
+    tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoder", use_auth_token=hf_token)
+    model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder", use_auth_token=hf_token, trust_remote_code=True)
     model.eval()
 
     device = "cuda" if torch.cuda.is_available() else "cpu"