Update model/download_model.py

model/download_model.py

@@ -1,29 +1,36 @@
-import requests
-import zipfile
+import os
+import subprocess
 
-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")
+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"]
 
 if not os.path.exists(MODEL_PATH):
-    os.makedirs(MODEL_DIR, exist_ok=True)
+    os.makedirs(MODEL_PATH)
+
+
+
+
+
+
+
+
+
 
-    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)
 
-    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
+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)
 
-    print("✅ Model đã tải thành công!")
-else:
-    print("✅ Model đã có sẵn, bỏ qua tải về.")
+print("Tải model thành công!")