fix stuff
This commit is contained in:
22
utils/vosk_setup.py
Normal file
22
utils/vosk_setup.py
Normal file
@@ -0,0 +1,22 @@
|
||||
import os
|
||||
import zipfile
|
||||
|
||||
import requests
|
||||
import vosk
|
||||
|
||||
from config import VOSK_MODEL_FOLDER, VOSK_MODEL_URL, VOSK_MODEL_ZIP
|
||||
|
||||
|
||||
def setup_vosk_model():
|
||||
if not os.path.isdir(VOSK_MODEL_FOLDER):
|
||||
print("Downloading Vosk model")
|
||||
response = requests.get(VOSK_MODEL_URL)
|
||||
with open(VOSK_MODEL_ZIP, "wb") as file:
|
||||
file.write(response.content)
|
||||
with zipfile.ZipFile(VOSK_MODEL_ZIP, 'r') as zip_ref:
|
||||
zip_ref.extractall()
|
||||
print("Downloaded!")
|
||||
|
||||
model = vosk.Model(VOSK_MODEL_FOLDER)
|
||||
rec = vosk.KaldiRecognizer(model, 16000)
|
||||
return rec
|
||||
Reference in New Issue
Block a user