Update model/download_model.py

model/download_model.py

@@ -2,9 +2,9 @@ import os
 import requests
 import zipfile
 
-MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-vn-0.4.zip"
+MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-en-us-0.22.zip"
 MODEL_DIR = "model"
-MODEL_NAME = "vosk-model-vn-0.4"  # Tên thư mục sau khi giải nén
+MODEL_NAME = "vosk-model-en-us-0.22"  # Đổi về model tiếng Anh
 MODEL_PATH = os.path.join(MODEL_DIR, MODEL_NAME)
 
 if not os.path.exists(MODEL_PATH):
@@ -12,7 +12,7 @@ if not os.path.exists(MODEL_PATH):
 
     zip_path = os.path.join(MODEL_DIR, "model.zip")
 
-    print("📥 Đang tải model Vosk tiếng Việt...")
+    print("📥 Đang tải model Vosk tiếng Anh...")
     response = requests.get(MODEL_URL, stream=True)
 
     if response.status_code != 200:
@@ -33,7 +33,6 @@ if not os.path.exists(MODEL_PATH):
     with zipfile.ZipFile(zip_path, "r") as zip_ref:
         zip_ref.extractall(MODEL_DIR)
 
-    # Kiểm tra xem thư mục giải nén có tồn tại không
     if os.path.exists(MODEL_PATH):
         print(f"✅ Model đã được giải nén tại: {MODEL_PATH}")
     else:
@@ -42,6 +41,6 @@ if not os.path.exists(MODEL_PATH):
 
     os.remove(zip_path)  # Xóa file zip sau khi giải nén
 
-    print("✅ Model tiếng Việt đã tải thành công!")
+    print("✅ Model tiếng Anh đã tải thành công!")
 else:
     print(f"✅ Model đã có sẵn tại {MODEL_PATH}, bỏ qua tải về.")