Update model/model_utils.py

model/model_utils.py

@@ -6,8 +6,8 @@ def load_model():
     model_name = "bigcode/starcoder"
     hf_token = os.getenv("HF_TOKEN")
 
-    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)
+    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"