wq2012 commited on
Commit
9793d07
·
verified ·
1 Parent(s): 2e8700d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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 = "Spoken Language Identification"
7
 
8
  description = """
9
- A demo of conformer-based spoken language identification. Paper: https://arxiv.org/abs/2202.12163
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(wav_file):
24
- top_lang, _ = runner.wav_to_lang(wav_file)
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,)