Update main.py

main.py

@@ -17,6 +17,7 @@ model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-128k-instruct
                                             device_map="auto",
                                             token=HF_TOKEN
                                             )
+tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True, token=HF_TOKEN)
 
 class Item(BaseModel):
     prompt: str