Update app.py

app.py

@@ -9,11 +9,11 @@ BATCH_SIZE = 1
 
 device = 0 if torch.cuda.is_available() else "cpu"
 
+"""
 mySpeechScore = SpeechScore([
         'PESQ','DNSMOS'
     ])
-
-
+"""
 # Copied from https://github.com/openai/whisper/blob/c09a7ae299c4c34c5839a76380ae407e7d785914/whisper/utils.py#L50
 def format_timestamp(seconds: float, always_include_hours: bool = False, decimal_marker: str = "."):
     if seconds is not None:
@@ -36,7 +36,13 @@ def format_timestamp(seconds: float, always_include_hours: bool = False, decimal
 
 
 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)
+    score_list=[]
+    if dnsmos:
+        score_list+=['DNSMOS']
+    if pesq:
+        score_list+=['PESQ']
+    mySpeechScore = SpeechScore(score_list)  
+    scores = mySpeechScore(test_path=test_file, reference_path=ref_file, window=None, score_rate=16000, return_mean=False)
     return scores