Create model/download_model.py

model/download_model.py

@@ -0,0 +1,23 @@
+import os
+import subprocess
+
+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_PATH)
+
+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("Tải model thành công!")