Update server.py

server.py

@@ -14,10 +14,10 @@ os.environ["HF_HOME"] = "/tmp"
 if not os.path.exists("/tmp"):
     os.makedirs("/tmp")
 
-# ✅ Load DeepSeek-Coder-V2-Base Model with correct cache directory
+# ✅ Load DeepSeek-Coder-V2-Base Model with `trust_remote_code=True`
 model_name = "deepseek-ai/DeepSeek-Coder-V2-Base"
-tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir="/tmp")
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", cache_dir="/tmp")
+tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir="/tmp", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", cache_dir="/tmp", trust_remote_code=True)
 
 class CodeRequest(BaseModel):
     user_story: str