Update app.py

app.py

@@ -129,13 +129,13 @@ if st.button('Сгенерировать потери'):
 
     data_lossy, samplerate = torchaudio.load('lossy.wav')
     data_enhanced, samplerate = torchaudio.load('enhanced.wav')
-    min_len = min(data_clean.shape[1], data_lossy.shape[1])
+    min_len = min(data_clean.shape[1], data_lossy.shape[1], data_enhanced.shape[1])
     data_clean = data_clean[:, :min_len]
     data_lossy = data_lossy[:, :min_len]
     data_enhanced = data_enhanced[:, :min_len]
 
 
-    stoi = STOI(48000)
+    stoi = STOI(samplerate)
     stoi_orig = round(float(stoi(data_clean, data_clean)),3)
     stoi_lossy = round(float(stoi(data_clean, data_lossy)),5)
     stoi_enhanced = round(float(stoi(data_clean, data_enhanced)),5)