update library

app.py

@@ -75,7 +75,6 @@ def speech2text(audio):
     return beam_search_output
 
 
-
 """Machine translation"""
 model_checkpoint = "huynguyen208/fantastic4-finetuned-vi-to-en-PhoMT-demo"
 translator = pipeline("translation", model=model_checkpoint)
@@ -88,15 +87,25 @@ def inference(audio):
     en_text = translate(vi_text)
     return en_text
 
+
+"""Gradio demo"""
+iface1 = gr.Interface(fn=translate,
+                     inputs=["text"],
+                     outputs="text",
+                     title = 'Translate Vietnamese to English',
+                     description = 'Mini Translator')
+iface1.launch(inline = False)
+
+
 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"
 examples=[['vi_speech_01.wav'], ['vi_speech_02.wav'], ['vi_speech_03.wav']]
-iface = gr.Interface(inference,
+iface2 = gr.Interface(inference,
             inputs,
             outputs, 
             title=title,
             description=description,
             examples=examples)
-iface.launch()
+iface2.launch()

requirements.txt

@@ -1,8 +1,10 @@
-torch
-torchaudio
+torch==1.9.0
+torchaudio==0.9.0
+transformers==4.9.2 
+datasets==1.11.0 
+pyctcdecode==v0.1.0
 speechbrain
 pydub
-datasets
 soundfile
 ffmpeg-python
 gradio