Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 from sidlingvo import wav_to_lang
 from huggingface_hub import hf_hub_download
+import numpy as np
 
 title = "Spoken Language Identification"
 
@@ -21,8 +22,9 @@ runner = wav_to_lang.WavToLangRunner(
     langid_model_file=os.path.join(model_path, "conformer_langid_medium.tflite"))
 
 def predict(wav_file):
-    top_lang, _ = runner.wav_to_lang(wav_file)
-    return "Predicted language: " + top_lang
+    top_lang, probs = runner.wav_to_lang(wav_file)
+    top_lang_prob = np.max(probs)
+    return "Predicted language: " + top_lang + "\nProbability: " + top_lang_prob
 
 if __name__ == "__main__":
     demo = gr.Interface(