Update app.py

app.py

@@ -34,14 +34,14 @@ def translate_to_english(prompt, lang_model_id, base_lang):
     if base_lang == "English":
         return prompt
     else:
-        text_output = text_to_text_generation(
+        output_text = text_to_text_generation(
             prompt=prompt,
             model_id=lang_model_id,
             device='cuda:0',
             target_lang='en'
         )
             
-        return text_output[0]
+        return output_text[0]
 
 
 def biogpt_text(
@@ -55,6 +55,7 @@ def biogpt_text(
     generator = pipeline("text-generation", model=biogpt_model_id, device="cuda:0")
     output = generator(en_prompt, max_length=250, num_return_sequences=1, do_sample=True)
     output = output[0]['generated_text']
+    
     if base_lang == "English":
         output_text = output
         
@@ -112,11 +113,11 @@ with app:
         with gr.Column():
             with gr.Tab("Text"):
                 input_text = gr.Textbox(lines=3, value="COVID-19 is", label="Text")
-                input_text_button = gr.Button(value="Predict")
                 input_biogpt_model = gr.Dropdown(choices=biogpt_model_list, value=biogpt_model_list[0], label='BioGpt Model')
                 input_m2m100_model = gr.Dropdown(choices=lang_model_list,  value=lang_model_list[1], label='Language Model')
                 input_base_lang = gr.Dropdown(lang_list, value="English", label="Base Language")
-            
+                input_text_button = gr.Button(value="Predict")
+
             with gr.Tab("Audio"):
                 input_audio = gr.Audio(source="microphone", type="filepath")
                 input_biogpt_model = gr.Dropdown(choices=biogpt_model_list, value=biogpt_model_list[0], label='BioGpt Model')
@@ -132,6 +133,6 @@ with app:
                 
     gr.Examples(examples, inputs=[input_text, input_biogpt_model, input_m2m100_model,input_base_lang], outputs=[prompt_text, output_text, translated_text], fn=biogpt_text, cache_examples=False)
     input_text_button.click(biogpt_text, inputs=[input_text, input_biogpt_model, input_m2m100_model,input_base_lang], outputs=[prompt_text, output_text, translated_text])
-    input_audio_button.click(biogpt_audio, inputs=[input_audio, input_biogpt_model,input_whisper_model,input_base_lang], outputs=[prompt_text, output_text, translated_text])
+    input_audio_button.click(biogpt_audio, inputs=[input_audio, input_biogpt_model, input_whisper_model,input_base_lang, input_m2m100_model], outputs=[prompt_text, output_text, translated_text])
     
 app.launch()