Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,10 +3,10 @@ import os
|
|
3 |
from sidlingvo import wav_to_lang
|
4 |
from huggingface_hub import hf_hub_download
|
5 |
|
6 |
-
title = "
|
7 |
|
8 |
description = """
|
9 |
-
A demo of conformer-based
|
10 |
"""
|
11 |
|
12 |
repo_id = "tflite-hub/conformer-lang-id"
|
@@ -20,14 +20,14 @@ runner = wav_to_lang.WavToLangRunner(
|
|
20 |
vad_mean_stddev_file=os.path.join(model_path, "vad_short_mean_stddev.csv"),
|
21 |
langid_model_file=os.path.join(model_path, "conformer_langid_medium.tflite"))
|
22 |
|
23 |
-
def predict(
|
24 |
-
top_lang, _ = runner.wav_to_lang(
|
25 |
return "Predicted language:" + top_lang
|
26 |
|
27 |
if __name__ == "__main__":
|
28 |
demo = gr.Interface(
|
29 |
fn=predict,
|
30 |
-
inputs=gr.Audio(type="filepath"),
|
31 |
outputs="text",
|
32 |
title=title,
|
33 |
description=description,)
|
|
|
3 |
from sidlingvo import wav_to_lang
|
4 |
from huggingface_hub import hf_hub_download
|
5 |
|
6 |
+
title = "Speaker Recognition Demo"
|
7 |
|
8 |
description = """
|
9 |
+
A demo of conformer-based speaker recognition. Paper: https://arxiv.org/abs/2104.02125
|
10 |
"""
|
11 |
|
12 |
repo_id = "tflite-hub/conformer-lang-id"
|
|
|
20 |
vad_mean_stddev_file=os.path.join(model_path, "vad_short_mean_stddev.csv"),
|
21 |
langid_model_file=os.path.join(model_path, "conformer_langid_medium.tflite"))
|
22 |
|
23 |
+
def predict(enroll_audio, test_audio):
|
24 |
+
top_lang, _ = runner.wav_to_lang(enroll_audio)
|
25 |
return "Predicted language:" + top_lang
|
26 |
|
27 |
if __name__ == "__main__":
|
28 |
demo = gr.Interface(
|
29 |
fn=predict,
|
30 |
+
inputs=[gr.Audio(type="filepath"), gr.Audio(type="filepath")],
|
31 |
outputs="text",
|
32 |
title=title,
|
33 |
description=description,)
|