Update app.py

app.py

@@ -10,7 +10,7 @@ BATCH_SIZE = 1
 device = 0 if torch.cuda.is_available() else "cpu"
 
 mySpeechScore = SpeechScore([
-        'PESQ'
+        'PESQ','DNSMOS'
     ])
 
 
@@ -35,8 +35,8 @@ def format_timestamp(seconds: float, always_include_hours: bool = False, decimal
         return seconds
 
 
-def score(test_file, ref_file, task, return_timestamps):
-    scores = mySpeechScore(test_path=test_file, reference_path=ref_file, window=None, score_rate=16000, return_mean=False)
+def score(test_file, ref_file, dnsmos, pesq, return_timestamps):
+    scores = mySpeechScore(test_path=test_file, reference_path=ref_file, dnsmos=dnsmos, pesq=pesq, window=None, score_rate=16000, return_mean=False)
     return scores
 
 
@@ -47,8 +47,9 @@ file_score = gr.Interface(
     inputs=[
         gr.Audio(sources=["upload"], label="test file", type="filepath"),
         gr.Audio(sources=["upload"], label="reference file", type="filepath"),
-        gr.Radio(["without reference", "with reference"], label="Task", info="choose non-instrusive or instrusive scoring"),
-        #gr.Checkbox(default=False, label="Return timestamps"),
+        #gr.Radio(["without reference", "with reference"], label="Task", info="choose non-instrusive or instrusive scoring"),
+        gr.Checkbox(default=False, label="DNSMOS"),
+        gr.Checkbox(default=False, label="PESQ"),
     ],
     outputs="text",
     #layout="horizontal",