alibabasglab commited on
Commit
761cc63
·
verified ·
1 Parent(s): a6f8f50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -9,7 +9,7 @@ BATCH_SIZE = 1
9
 
10
  device = 0 if torch.cuda.is_available() else "cpu"
11
 
12
- def score(test_file, ref_file, score_list_nis, score_list_is, return_timestamps):
13
  score_list = []
14
  if score_list_nis is not None:
15
  score_list += score_list_nis
@@ -19,16 +19,23 @@ def score(test_file, ref_file, score_list_nis, score_list_is, return_timestamps)
19
  scores = mySpeechScore(test_path=test_file, reference_path=ref_file, window=None, score_rate=16000, return_mean=False)
20
  return scores
21
 
 
 
 
 
 
 
 
22
 
23
  demo = gr.Blocks()
24
 
25
  file_score = gr.Interface(
26
- fn=score,
27
  inputs=[
28
  gr.Audio(sources=["upload"], label="test file", type="filepath"),
29
  gr.Audio(sources=["upload"], label="reference file", type="filepath"),
30
  gr.Dropdown(
31
- ["DNSMOS", "SRMR"], value=["DNSMOS", "SRMR"], multiselect=True, label="Non-Intrusive Scores", info="Choose scores to include, no reference audio is required."
32
  ),
33
  gr.Dropdown(
34
  ["PESQ", 'NB_PESQ', 'STOI', 'SISDR',
@@ -47,7 +54,7 @@ file_score = gr.Interface(
47
  )
48
 
49
  mic_score = gr.Interface(
50
- fn=score,
51
  inputs=[
52
  gr.Audio(sources=["microphone"],
53
  waveform_options=gr.WaveformOptions(
 
9
 
10
  device = 0 if torch.cuda.is_available() else "cpu"
11
 
12
+ def score_file(test_file, ref_file, score_list_nis, score_list_is):
13
  score_list = []
14
  if score_list_nis is not None:
15
  score_list += score_list_nis
 
19
  scores = mySpeechScore(test_path=test_file, reference_path=ref_file, window=None, score_rate=16000, return_mean=False)
20
  return scores
21
 
22
+ def score_mic(test_file, score_list_nis):
23
+ score_list = []
24
+ if score_list_nis is not None:
25
+ score_list += score_list_nis
26
+ mySpeechScore = SpeechScore(score_list)
27
+ scores = mySpeechScore(test_path=test_file, reference_path=ref_file, window=None, score_rate=16000, return_mean=False)
28
+ return scores
29
 
30
  demo = gr.Blocks()
31
 
32
  file_score = gr.Interface(
33
+ fn=score_file,
34
  inputs=[
35
  gr.Audio(sources=["upload"], label="test file", type="filepath"),
36
  gr.Audio(sources=["upload"], label="reference file", type="filepath"),
37
  gr.Dropdown(
38
+ ["DNSMOS", "SRMR"], value=["DNSMOS"], multiselect=True, label="Non-Intrusive Scores", info="Choose scores to include, no reference audio is required."
39
  ),
40
  gr.Dropdown(
41
  ["PESQ", 'NB_PESQ', 'STOI', 'SISDR',
 
54
  )
55
 
56
  mic_score = gr.Interface(
57
+ fn=score_mic,
58
  inputs=[
59
  gr.Audio(sources=["microphone"],
60
  waveform_options=gr.WaveformOptions(