Update app.py

app.py

@@ -131,8 +131,7 @@ def translation(source_lang, target_lang):
     tokenizer.src_lang = src_lang
     with torch.no_grad():
         encoded_input = tokenizer(user_input, return_tensors="pt").to(device)
-        generated_tokens = model.generate(
-        **encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
+        generated_tokens = translation_model.generate(encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
         translated_text = tokenizer.batch_decode(
         generated_tokens, skip_special_tokens=True)[0]
     return translated_text
@@ -143,6 +142,7 @@ with demo:
   audio = gr.Audio(type="filepath", label = "Upload a file")
   text0 = gr.Textbox()
   text = gr.Textbox()
+  text2 = gr.Textbox()
   source_lang = gr.Dropdown(label="Source lang", choices=list(lang_id.keys()))
   target_lang = gr.Dropdown(label="target lang", choices=list(lang_id.keys()))
    
@@ -151,7 +151,7 @@ with demo:
     #                source_lang])
   b1 = gr.Button("convert to text")
   b3 = gr.Button("translate")
-  b3.click(translation, inputs = text0, outputs = text)
+  b3.click(translation, inputs = text0, outputs = text2)
   b1.click(audio_a_text, inputs=audio, outputs=text)
 
   b2 = gr.Button("Classification of speech")