Update app.py

app.py

@@ -125,7 +125,9 @@ lang_id = {
     "Chinese": "zh",
     "Zulu": "zu",
 }
-def translation_text(src_lang, trg_lang, user_input):
+def translation_text(source_lang, target_lang, user_input):
+    src_lang = lang_id[source_lang]
+    trg_lang = lang_id[target_lang]
     tokenizer.src_lang = src_lang
     with torch.no_grad():
         encoded_input = tokenizer(user_input, return_tensors="pt").to(device)
@@ -146,14 +148,13 @@ with demo:
   text2 = gr.Textbox()
   source_lang = gr.Dropdown(label="Source lang", choices=list(lang_id.keys()), value=list(lang_id.keys())[0])
   target_lang = gr.Dropdown(label="target lang", choices=list(lang_id.keys()), value=list(lang_id.keys())[0])
-  #src_lang = lang_id[source_lang]
-  #rg_lang = lang_id[target_lang]
+  
   #gr.Examples(examples = list(lang_id.keys()),
    #             inputs=[
     #                source_lang])
   b1 = gr.Button("convert to text")
   b3 = gr.Button("translate")
-  #b3.click(translation_text, inputs = (src_lang, trg_lang, text0), outputs = text2)
+  b3.click(translation_text, inputs = [source_lang, target_lang, text0], outputs = text2)
   b1.click(audio_a_text, inputs=audio, outputs=text)
 
   b2 = gr.Button("Classification of speech")