Upload app.py

app.py

@@ -27,7 +27,8 @@ MODEL_B_PATH = "saved_models/best_model_B.pth"
 FUZZY_RULES_PATH = "data/datafuzzy29d.csv"
 
 # --- Tải Mô hình STT (ĐÃ THAY ĐỔI SANG WHISPER) ---
-STT_MODEL_ID = "vi-openai/whisper-base-vi"
+# === SỬA LỖI LẦN 2: Đã cập nhật tên model chính xác ===
+STT_MODEL_ID = "vinai/vinai-whisper-base" 
 print(f"Đang tải mô hình STT Whisper: {STT_MODEL_ID}...")
 # Cần 'language' và 'task' để bộ xử lý biết cách hoạt động
 audio_processor = WhisperProcessor.from_pretrained(STT_MODEL_ID, language="vi", task="transcribe")
@@ -179,7 +180,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[text_out, pred_A_out, pred_B_out, final_pred_out]
     )
     
-    gr.Markdown("Lưu ý: Mô hình STT hiện đang sử dụng `trituenhantao/whisper-base-vi`.")
+    gr.Markdown("Lưu ý: Mô hình STT hiện đang sử dụng `vinai/vinai-whisper-base`.")
 
 print("Đang khởi chạy demo...")
 demo.launch()
+