tudeplom commited on
Commit
d102cd2
·
verified ·
1 Parent(s): 68c9c91

Update model/download_model.py

Browse files
Files changed (1) hide show
  1. model/download_model.py +2 -2
model/download_model.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import requests
3
  import zipfile
4
 
5
- MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-small-en-us-0.15.zip"
6
  MODEL_DIR = "model"
7
  MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model")
8
 
@@ -21,7 +21,7 @@ if not os.path.exists(MODEL_PATH):
21
  with zipfile.ZipFile(zip_path, "r") as zip_ref:
22
  zip_ref.extractall(MODEL_DIR)
23
 
24
- os.rename(os.path.join(MODEL_DIR, "vosk-model-small-en-us-0.15"), MODEL_PATH)
25
  os.remove(zip_path) # Xóa file zip sau khi giải nén
26
 
27
  print("✅ Model đã tải thành công!")
 
2
  import requests
3
  import zipfile
4
 
5
+ MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-vn-0.4.zip"
6
  MODEL_DIR = "model"
7
  MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model")
8
 
 
21
  with zipfile.ZipFile(zip_path, "r") as zip_ref:
22
  zip_ref.extractall(MODEL_DIR)
23
 
24
+ os.rename(os.path.join(MODEL_DIR, "vosk-model-vn-0.4"), MODEL_PATH)
25
  os.remove(zip_path) # Xóa file zip sau khi giải nén
26
 
27
  print("✅ Model đã tải thành công!")