Update app.py

app.py

@@ -20,6 +20,8 @@ from torchmetrics.audio import ShortTimeObjectiveIntelligibility as STOI
 from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality as PESQ
 
 
+from plc_mos import PLCMOSEstimator
+
 
 @st.cache
 def load_model():
@@ -228,8 +230,17 @@ if st.button('Сгенерировать потери'):
 
 
 
-    
 
+
+    PLC_example=PLCMOSEstimator()
+    PLC_org = PLC_example.run(audio_degraded=data_clean, audio_clean=data_clean)[0]
+    PLC_lossy = PLC_example.run(audio_degraded=data_lossy, audio_clean=data_clean)[0]
+    PLC_enhanced = PLC_example.run(audio_degraded=data_enhanced, audio_clean=data_clean)[0]
+
+    PLC_mass = [PLC_org, PLC_lossy, PLC_enhanced]
+
+
+    
     df = pd.DataFrame(columns=['Audio', 'PESQ', 'STOI', 'PLCMOS', 'LSD'])
 
     df['Audio'] = ['Clean', 'Lossy', 'Enhanced']
@@ -240,6 +251,8 @@ if st.button('Сгенерировать потери'):
 
     df['LSD'] = lsd_mass
 
+    df['PLCMOS'] = PLC_mass
+
     st.table(df)