Update app.py

app.py

@@ -29,8 +29,8 @@ from jiwer import wer
 
 
 @st.cache
-def load_model():
-    path = 'lightning_logs/version_0/checkpoints/frn_modified.onnx'
+def load_model(model):
+    path = 'lightning_logs/version_0/checkpoints/' + str(model)
     onnx_model = onnx.load(path)
     options = onnxruntime.SessionOptions()
     options.intra_op_num_threads = 2
@@ -114,6 +114,13 @@ target = target[:packet_size * (len(target) // packet_size)]
 st.text('Ваше аудио')
 st.audio(uploaded_file)
 
+option = st.selectbox(
+     '1 or 2 onnx?',
+     ('frn.onnx', 'frn_modified.onnx'))
+
+st.write('You selected:', option)
+
+
 st.subheader('2. Выберите желаемый процент потерь')
 slider = [st.slider("Ожидаемый процент потерь для генератора потерь цепи Маркова", 0, 100, step=1)]
 loss_percent = float(slider[0])/100
@@ -126,7 +133,9 @@ hann = torch.sqrt(torch.hann_window(window))
 lossy_input_tensor = torch.tensor(lossy_input)
 re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
     1).numpy().astype(np.float32)
-session, onnx_model, input_names, output_names = load_model()
+
+
+session, onnx_model, input_names, output_names = load_model(option)
 
 if st.button('Сгенерировать потери'):
     with st.spinner('Ожидайте...'):