Spaces:
Running
Running
Yurii Paniv
commited on
Commit
·
a5c7abb
1
Parent(s):
8451e68
Add scorer
Browse files- .gitignore +2 -1
- Dockerfile +1 -1
- client.py +1 -6
- templates/hello.html +0 -1
.gitignore
CHANGED
|
@@ -131,4 +131,5 @@ dmypy.json
|
|
| 131 |
*.tflite
|
| 132 |
.DS_Store
|
| 133 |
|
| 134 |
-
/data
|
|
|
|
|
|
| 131 |
*.tflite
|
| 132 |
.DS_Store
|
| 133 |
|
| 134 |
+
/data
|
| 135 |
+
*.scorer
|
Dockerfile
CHANGED
|
@@ -4,6 +4,6 @@ WORKDIR /app
|
|
| 4 |
RUN apt-get update
|
| 5 |
RUN apt-get install -y ffmpeg
|
| 6 |
RUN wget https://github.com/robinhad/voice-recognition-ua/releases/download/v0.4/uk.tflite
|
| 7 |
-
RUN wget https://github.com/
|
| 8 |
RUN pip install -r requirements.txt
|
| 9 |
CMD uwsgi app.ini --http 0.0.0.0:$PORT
|
|
|
|
| 4 |
RUN apt-get update
|
| 5 |
RUN apt-get install -y ffmpeg
|
| 6 |
RUN wget https://github.com/robinhad/voice-recognition-ua/releases/download/v0.4/uk.tflite
|
| 7 |
+
RUN wget https://github.com/robinhad/voice-recognition-ua/releases/download/v0.4/kenlm.scorer
|
| 8 |
RUN pip install -r requirements.txt
|
| 9 |
CMD uwsgi app.ini --http 0.0.0.0:$PORT
|
client.py
CHANGED
|
@@ -14,14 +14,9 @@ def client(audio_file, lang="uk"):
|
|
| 14 |
model_load_start = timer()
|
| 15 |
# sphinx-doc: python_ref_model_start
|
| 16 |
model = "./uk.tflite"
|
| 17 |
-
if lang not in ["en", "uk"]:
|
| 18 |
-
lang = "uk"
|
| 19 |
-
if lang == "uk":
|
| 20 |
-
model = "./uk.tflite"
|
| 21 |
-
if lang == "en":
|
| 22 |
-
model = "./deepspeech-0.9.1-models.tflite"
|
| 23 |
|
| 24 |
ds = Model(model)
|
|
|
|
| 25 |
# sphinx-doc: python_ref_model_stop
|
| 26 |
model_load_end = timer() - model_load_start
|
| 27 |
print('Loaded model in {:.3}s.'.format(model_load_end), file=sys.stderr)
|
|
|
|
| 14 |
model_load_start = timer()
|
| 15 |
# sphinx-doc: python_ref_model_start
|
| 16 |
model = "./uk.tflite"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
ds = Model(model)
|
| 19 |
+
ds.enableExternalScorer("kenlm.scorer")
|
| 20 |
# sphinx-doc: python_ref_model_stop
|
| 21 |
model_load_end = timer() - model_load_start
|
| 22 |
print('Loaded model in {:.3}s.'.format(model_load_end), file=sys.stderr)
|
templates/hello.html
CHANGED
|
@@ -22,7 +22,6 @@
|
|
| 22 |
<div class="col-5 col-sm-3">
|
| 23 |
<select id="lang" class="browser-default custom-select" style="width: 100%">
|
| 24 |
<option selected value="uk">Українська</option>
|
| 25 |
-
<option value="en">Англійська</option>
|
| 26 |
</select>
|
| 27 |
</div>
|
| 28 |
</div>
|
|
|
|
| 22 |
<div class="col-5 col-sm-3">
|
| 23 |
<select id="lang" class="browser-default custom-select" style="width: 100%">
|
| 24 |
<option selected value="uk">Українська</option>
|
|
|
|
| 25 |
</select>
|
| 26 |
</div>
|
| 27 |
</div>
|