23 lines
600 B
Python
23 lines
600 B
Python
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
|