Update app.py

app.py

@@ -18,13 +18,6 @@ import torchaudio
 
 from torch_pesq import PesqLoss
 
-pesq = PesqLoss(0.5,
-    sample_rate=44100, 
-)
-
-mos = pesq.mos(reference, degraded)
-loss = pesq(reference, degraded)
-
 
 @st.cache
 def load_model():
@@ -154,16 +147,25 @@ if st.button('Сгенерировать потери'):
     stoi_mass=[stoi_orig, stoi_lossy, stoi_enhanced]
 
 
+
+    
     #if samplerate != 16000:
     #    data_lossy = librosa.resample(data_lossy, orig_sr=48000, target_sr=16000)
     #    data_clean = librosa.resample(data_clean, orig_sr=48000, target_sr=16000)
     #    data_enhanced = librosa.resample(data_enhanced, orig_sr=48000, target_sr=16000)
-#
+    #
+
+    pesq = PesqLoss(0.5, sample_rate=48000)
+
+    pesq_orig = pesq.mos(data_clean, data_clean)
+    pesq_lossy = pesq.mos(data_clean, data_lossy)
+    pesq_enhanced= pesq.mos(data_clean, data_enhanced)
+
     #pesq_orig = pesq(fs = 16000, ref = data_clean, deg = data_clean, mode='nb')
     #pesq_lossy = pesq(fs = 16000, ref = data_clean, deg = data_lossy, mode='nb')
     #pesq_enhanced = pesq(fs = 16000, ref = data_clean, deg = data_enhanced, mode='nb')
 
-    #psq_mas=[pesq_orig, pesq_lossy, pesq_enhanced]
+    psq_mas=[pesq_orig, pesq_lossy, pesq_enhanced]
 
 
 
@@ -171,7 +173,7 @@ if st.button('Сгенерировать потери'):
 
     df['Audio'] = ['Clean', 'Lossy', 'Enhanced']
 
-    #df['PESQ'] = psq_mas
+    df['PESQ'] = psq_mas
 
     df['STOI'] = stoi_mass