Update model/download_model.py

model/download_model.py

@@ -1,23 +1,29 @@
 import os
-import subprocess
+import requests
+import zipfile
 
-MODEL_PATH = "model"
-MODEL_URL = "https://huggingface.co/vosk-models/vosk-small-en-us/resolve/main"
-
-# Danh sách file cần tải
-files = ["am", "conf", "graph", "ivector", "rescore"]
+MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-small-en-us-0.15.zip"
+MODEL_DIR = "model"
+MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model")
 
 if not os.path.exists(MODEL_PATH):
-    os.makedirs(MODEL_PATH)
+    os.makedirs(MODEL_DIR, exist_ok=True)
+
+    zip_path = os.path.join(MODEL_DIR, "model.zip")
+
+    print("📥 Đang tải model Vosk...")
+    response = requests.get(MODEL_URL, stream=True)
+    with open(zip_path, "wb") as f:
+        for chunk in response.iter_content(chunk_size=8192):
+            f.write(chunk)
+
+    print("📦 Giải nén model...")
+    with zipfile.ZipFile(zip_path, "r") as zip_ref:
+        zip_ref.extractall(MODEL_DIR)
 
-for file in files:
-    file_url = f"{MODEL_URL}/{file}.zip"
-    file_path = os.path.join(MODEL_PATH, f"{file}.zip")
-    
-    if not os.path.exists(file_path):
-        print(f"Đang tải {file_url} ...")
-        subprocess.run(["wget", "-O", file_path, file_url])
-        subprocess.run(["unzip", "-o", file_path, "-d", MODEL_PATH])
-        os.remove(file_path)
+    os.rename(os.path.join(MODEL_DIR, "vosk-model-small-en-us-0.15"), MODEL_PATH)
+    os.remove(zip_path)  # Xóa file zip sau khi giải nén
 
-print("Tải model thành công!")
+    print("✅ Model đã tải thành công!")
+else:
+    print("✅ Model đã có sẵn, bỏ qua tải về.")