Update app.py

app.py

@@ -3,10 +3,10 @@ import os
 from sidlingvo import wav_to_lang
 from huggingface_hub import hf_hub_download
 
-title = "Spoken Language Identification"
+title = "Speaker Recognition Demo"
 
 description = """
-A demo of conformer-based spoken language identification. Paper: https://arxiv.org/abs/2202.12163
+A demo of conformer-based speaker recognition. Paper: https://arxiv.org/abs/2104.02125
 """
 
 repo_id = "tflite-hub/conformer-lang-id"
@@ -20,14 +20,14 @@ runner = wav_to_lang.WavToLangRunner(
     vad_mean_stddev_file=os.path.join(model_path, "vad_short_mean_stddev.csv"),
     langid_model_file=os.path.join(model_path, "conformer_langid_medium.tflite"))
 
-def predict(wav_file):
-    top_lang, _ = runner.wav_to_lang(wav_file)
+def predict(enroll_audio, test_audio):
+    top_lang, _ = runner.wav_to_lang(enroll_audio)
     return "Predicted language:" + top_lang
 
 if __name__ == "__main__":
     demo = gr.Interface(
         fn=predict,
-        inputs=gr.Audio(type="filepath"),
+        inputs=[gr.Audio(type="filepath"), gr.Audio(type="filepath")],
         outputs="text",
         title=title,
         description=description,)