Update main.py

main.py

@@ -18,7 +18,7 @@ if HF_TOKEN:
     login(HF_TOKEN)
 
 # Load AI Model
-ai_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+ai_model = pipeline("text-generation", model="google/flan-t5-base")
 
 app = FastAPI()
 
@@ -81,7 +81,7 @@ async def ai_personal_advice(request: AIRequest):
     Hãy tư vấn cách cải thiện hiệu suất làm việc dựa trên thông tin trên.
     """
     
-    response = ai_model(prompt, max_length=300)
+    response = ai_model(prompt, max_length=200)
     return {"advice": response[0]["generated_text"]}
 
 # Trang chủ