Handle rename to Mozilla Voice STT

This commit is contained in:
Reuben Morais 2020-08-02 17:27:21 +02:00
Родитель 5387b6dd23
Коммит 19797e0e59
3 изменённых файлов: 6 добавлений и 6 удалений

Просмотреть файл

@ -1,7 +1,7 @@
import time, logging
from datetime import datetime
import threading, collections, queue, os, os.path
import deepspeech
import mozilla_voice_stt as stt
import numpy as np
import pyaudio
import wave
@ -160,7 +160,7 @@ def main(ARGS):
print('Initializing model...')
logging.info("ARGS.model: %s", ARGS.model)
model = deepspeech.Model(ARGS.model)
model = stt.Model(ARGS.model)
if ARGS.scorer:
logging.info("ARGS.scorer: %s", ARGS.scorer)
model.enableExternalScorer(ARGS.scorer)

Просмотреть файл

@ -14,10 +14,10 @@ get_python_wheel_url()
extract_python_versions "${this_python_version}" "pyver" "pyver_pkg" "py_unicode_type" "pyconf" "pyalias"
echo "$(get_python_pkg_url "${pyver_pkg}" "${py_unicode_type}" "deepspeech" https://community-tc.services.mozilla.com/api/queue/v1/task/${DEP_TASK_ID}/artifacts/public)"
echo "$(get_python_pkg_url "${pyver_pkg}" "${py_unicode_type}" "mozilla_voice_stt" https://community-tc.services.mozilla.com/api/queue/v1/task/${DEP_TASK_ID}/artifacts/public)"
}
get_npm_package_url()
{
echo "https://community-tc.services.mozilla.com/api/queue/v1/task/${DEP_TASK_ID}/artifacts/public/deepspeech-${DS_VERSION}.tgz"
echo "https://community-tc.services.mozilla.com/api/queue/v1/task/${DEP_TASK_ID}/artifacts/public/mozilla_voice_stt-${DS_VERSION}.tgz"
}

Просмотреть файл

@ -2,7 +2,7 @@ import glob
import webrtcvad
import logging
import wavSplit
from deepspeech import Model
import mozilla_voice_stt
from timeit import default_timer as timer
'''
@ -15,7 +15,7 @@ Returns a list [DeepSpeech Object, Model Load Time, Scorer Load Time]
'''
def load_model(models, scorer):
model_load_start = timer()
ds = Model(models)
ds = mozilla_voice_stt.Model(models)
model_load_end = timer() - model_load_start
logging.debug("Loaded model in %0.3fs." % (model_load_end))