Update main.py

main.py

@@ -19,8 +19,8 @@ if HF_TOKEN:
     login(HF_TOKEN)
 
 # Load AI Model
-tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small",token=HF_TOKEN)
-ai_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small",token=HF_TOKEN)
+tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-large",token=HF_TOKEN)
+ai_model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-large",token=HF_TOKEN)
 
 app = FastAPI()