Update app.py

app.py

@@ -114,11 +114,17 @@ target = target[:packet_size * (len(target) // packet_size)]
 st.text('Ваше аудио')
 st.audio(uploaded_file)
 
-option = st.selectbox(
-     '1 or 2 onnx?',
+model_ver = st.selectbox(
+     'Оригинал или Pruned ?',
      ('frn.onnx', 'frn_modified.onnx'))
 
-st.write('You selected:', option)
+st.write('Вы выбрали:', model_ver)
+
+lang = st.selectbox(
+     'Выберите язык',
+     ('ru-RU', 'en-EN'))
+
+st.write('Вы выбрали:', lang)
 
 
 st.subheader('2. Выберите желаемый процент потерь')
@@ -312,7 +318,7 @@ if st.button('Сгенерировать потери'):
     with harvard as source:
         audio = r.record(source)
 
-    orig = r.recognize_google(audio, language = "en-EN")
+    orig = r.recognize_google(audio, language = str(lang))
 
 
 
@@ -327,7 +333,7 @@ if st.button('Сгенерировать потери'):
     try:
         with harvard as source:
             audio = r.record(source)
-        lossy = r.recognize_google(audio, language = "en-EN")
+        lossy = r.recognize_google(audio, language = str(lang))
         #print("Распознанный текст:", text)
     except speech_r.UnknownValueError:
         #st.text("Система не смогла распознать аудио")
@@ -347,7 +353,7 @@ if st.button('Сгенерировать потери'):
     try:
         with harvard as source:
             audio = r.record(source)
-        enhanced = r.recognize_google(audio, language = "en-EN")
+        enhanced = r.recognize_google(audio, language = str(lang))
         #print("Распознанный текст:", text)
     except speech_r.UnknownValueError:
         #st.text("Система не смогла распознать улучшенное аудио")