Update app.py

app.py

@@ -24,7 +24,7 @@ os.makedirs(TEMP_DIR, exist_ok=True)
 # Mô hình sử dụng
 STT_MODEL = "openai/whisper-tiny.en"
 TTS_MODEL = "facebook/mms-tts-eng"
-CHAT_MODEL = "mistralai/Mistral-7B-Instruct-v0.1"  # Cập nhật model hợp lệ
+CHAT_MODEL = "mistralai/Mistral-7B-Instruct-v0.1"  # Model nhẹ hơn
 
 # Giao diện HTML
 HTML_CONTENT = """
@@ -64,7 +64,7 @@ async def text_to_speech(text: str) -> str:
     output_path = os.path.join(TEMP_DIR, "output.wav")
     try:
         print(f"🔄 Đang xử lý TTS với model {TTS_MODEL}...")
-        response = client.text_to_speech(model=TTS_MODEL, text=text)
+        response = client.text_to_speech(TTS_MODEL, text)
         with open(output_path, "wb") as f:
             f.write(response)
         with open(output_path, "rb") as f:
@@ -78,7 +78,7 @@ async def speech_to_text(file: UploadFile) -> str:
     try:
         print(f"🔄 Đang xử lý STT với model {STT_MODEL}...")
         audio_data = await file.read()
-        response = client.automatic_speech_recognition(model=STT_MODEL, audio=audio_data, return_timestamps=False)
+        response = client.automatic_speech_recognition(STT_MODEL, audio_data)
         return response.get("text", "Không nghe được gì.")
     except Exception as e:
         print(f"❌ Lỗi STT: {e}")
@@ -87,7 +87,7 @@ async def speech_to_text(file: UploadFile) -> str:
 async def generate_text(prompt: str) -> str:
     try:
         print(f"🔄 Đang xử lý hội thoại với model {CHAT_MODEL}...")
-        response = client.text_generation(model=CHAT_MODEL, prompt=prompt, max_new_tokens=100)
+        response = client.text_generation(CHAT_MODEL, prompt, max_new_tokens=100)
         return response.get("generated_text", "Xin lỗi, tôi gặp lỗi khi xử lý câu hỏi của bạn.").strip()
     except Exception as e:
         print(f"❌ Lỗi xử lý hội thoại: {e}")
@@ -113,4 +113,4 @@ async def chatbot(prompt: str = Form(None), file: UploadFile = None):
 
 if __name__ == "__main__":
     print("🚀 Khởi động FastAPI Server...")
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)