update speech2text module

app.py

@@ -106,16 +106,16 @@ with gr.Blocks() as demo:
             gr.Examples(examples=vi_example_text,
                         inputs=[vietnamese])
         with gr.TabItem("Speech2text and translation"):
-            inputs = gr.inputs.Audio(label="Input Audio", type="file")
-            outputs =  gr.outputs.Textbox(label="Output Text")
-            title = "Speech to text and translate Vietnamese to English"
-            description = "Gradio demo for a wav2vec2-base-vietnamese-250h and Helsinki-NLP/opus-mt-vi-en"
-            gr.Interface(inference,
-                        inputs,
-                        outputs, 
-                        title=title,
-                        description=description,
-                        examples=vi_example_voice)
+            with gr.Row():
+                with gr.Column():
+                    audio = gr.Audio(source="microphone", label="Input Audio", type="filepath")
+                    translate_button = gr.Button(value="Translate To English")
+                with gr.Column():
+                    enlish = gr.Textbox(label="English Text")
+                
+            translate_button.click(lambda voice: inference(voice), inputs=audio, outputs=english)
+            gr.Examples(examples=vi_example_voice,
+                        inputs=[audio])
 
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -5,6 +5,8 @@ datasets==1.11.0
 pyctcdecode==v0.1.0
 speechbrain
 pydub
+kenlm
+pyctcdecode
 soundfile
 ffmpeg-python
 gradio