Update app.py

app.py

@@ -11,10 +11,21 @@ import onnxruntime, onnx
 import matplotlib.pyplot as plt
 from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
 from pystoi import stoi
-from pesq import pesq
+#from pesq import pesq
 import pandas as pd
 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():
     path = 'lightning_logs/version_0/checkpoints/frn.onnx'