Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ BATCH_SIZE = 1
|
|
9 |
|
10 |
device = 0 if torch.cuda.is_available() else "cpu"
|
11 |
|
12 |
-
def
|
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=
|
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"
|
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=
|
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(
|