Update model/download_model.py

model/download_model.py

@@ -4,7 +4,8 @@ import zipfile
 
 MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-vn-0.4.zip"
 MODEL_DIR = "model"
-MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model-vn-0.4")  # Đổi lại cho đúng với thư mục sau khi giải nén
+MODEL_NAME = "vosk-model-vn-0.4"  # Tên thư mục sau khi giải nén
+MODEL_PATH = os.path.join(MODEL_DIR, MODEL_NAME)
 
 if not os.path.exists(MODEL_PATH):
     os.makedirs(MODEL_DIR, exist_ok=True)
@@ -33,9 +34,8 @@ if not os.path.exists(MODEL_PATH):
         zip_ref.extractall(MODEL_DIR)
 
     # Kiểm tra xem thư mục giải nén có tồn tại không
-    extracted_path = os.path.join(MODEL_DIR, "vosk-model-vn-0.4")
-    if os.path.exists(extracted_path):
-        print(f"✅ Model đã được giải nén tại: {extracted_path}")
+    if os.path.exists(MODEL_PATH):
+        print(f"✅ Model đã được giải nén tại: {MODEL_PATH}")
     else:
         print(f"❌ Lỗi: Model không tồn tại sau khi giải nén! Kiểm tra lại.")
         exit(1)