Update app.py

modified return statement

app.py

@@ -26,6 +26,14 @@ def predict(url=None, translation="No",tran_lang="en"):
 
     _,probs = model_m.detect_language(mel)
 
+    p = -1
+    for key in probs:
+        if probs[key] >p:
+            p = probs[key]
+    for keys in probs:
+        if probs[keys] == p:
+            detected_lang = keys
+
     lang_dict = sorted(probs)
     video_url = url
     v_data = pytube.YouTube(video_url)
@@ -36,7 +44,7 @@ def predict(url=None, translation="No",tran_lang="en"):
 
     if translation == "Yes":
         trans = model_m.transcribe(test_audio_file,language=tran_lang,fp16=False)["text"]
-        return lang_dict, transcription, trans
+        return detected_lang, transcription, trans
     else:
         return lang_dict, transcription
 
@@ -51,7 +59,7 @@ else:
 if st.button("Generate"):
     lang_d,transcription,trans = predict(url,translation=tran_req,tran_lang=lang)
 
-    st.write(lang_d.keys(),lang_d.values())
+    st.write(lang_d)
     canvas = Canvas("transcript.pdf")
     canvas.drawString(72, 72, transcription)
     canvas.save()