Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ import speech_recognition as speech_r
|
|
28 |
from jiwer import wer
|
29 |
import time
|
30 |
|
31 |
-
@st.
|
32 |
def load_model(model):
|
33 |
path = 'lightning_logs/version_0/checkpoints/' + str(model)
|
34 |
onnx_model = onnx.load(path)
|
@@ -406,7 +406,17 @@ if st.button('Сгенерировать потери'):
|
|
406 |
|
407 |
df_1['WER'] = WER_mass
|
408 |
|
409 |
-
st.dataframe(df_1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
410 |
|
411 |
st.bar_chart(df_1, x="Audio", y="PESQ")
|
412 |
st.bar_chart(df_1, x="Audio", y="STOI")
|
|
|
28 |
from jiwer import wer
|
29 |
import time
|
30 |
|
31 |
+
@st.cache_data
|
32 |
def load_model(model):
|
33 |
path = 'lightning_logs/version_0/checkpoints/' + str(model)
|
34 |
onnx_model = onnx.load(path)
|
|
|
406 |
|
407 |
df_1['WER'] = WER_mass
|
408 |
|
409 |
+
st.dataframe(df_1, column_config={
|
410 |
+
"PESQ": st.column_config.NumberColumn(
|
411 |
+
help="Перцептивная оценка качества речи - https://ieeexplore.ieee.org/document/941023"),
|
412 |
+
"STOI": st.column_config.NumberColumn(
|
413 |
+
help="Индекс объективной кратковременной разборчивости - https://ieeexplore.ieee.org/document/5495701"),
|
414 |
+
"PLCMOSv1": st.column_config.NumberColumn(
|
415 |
+
help="Эталонная метрика PLCMOS - https://arxiv.org/abs/2305.15127"),
|
416 |
+
"PLCMOSv2": st.column_config.NumberColumn(
|
417 |
+
help="Неэталонная метрика PLCMOS - https://arxiv.org/abs/2305.15127"),
|
418 |
+
"WER": st.column_config.NumberColumn(
|
419 |
+
help="Процент нераспознанных слов - https://deepgram.com/learn/what-is-word-error-rate")})
|
420 |
|
421 |
st.bar_chart(df_1, x="Audio", y="PESQ")
|
422 |
st.bar_chart(df_1, x="Audio", y="STOI")
|