Add trust_remote_code=True for DeepSeek model

app.py

@@ -6,8 +6,12 @@ import torch
 # Load the model and tokenizer
 MODEL_NAME = "deepseek-ai/DeepSeek-V3-Base"  # Change to the model you want
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto").to(device)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_NAME,
+    device_map="auto",
+    trust_remote_code=True  # Allow execution of custom code
+).to(device)
 
 app = FastAPI()