| import gradio as gr | |
| from transformers import pipeline | |
| s2t=gr.Interface.load('huggingface/facebook/s2t-medium-librispeech-asr') | |
| grammar = gr.Interface.load('huggingface/prithivida/grammar_error_correcter_v1') | |
| def out(audio1,audio2): | |
| if (audio1==None) and (audio2==None): | |
| return "no audio","no audio" | |
| elif audio1==None: | |
| x=s2t(audio2) | |
| return x, grammar(x) | |
| else: | |
| x=s2t(audio1) | |
| return x, grammar(x) | |
| iface = gr.Interface( | |
| fn=out, | |
| title="Speech Audio to text (with corrected grammar)", | |
| description="2 possible input methods of audio file, transformed to text (as an output) and corrected grammar after(another output)!", | |
| inputs=[gr.inputs.Audio(source="upload", type="filepath", label=None, optional=True), | |
| gr.inputs.Audio(source="microphone", type="filepath", label=None, optional=True)], | |
| examples=[["Grammar-Correct-Sample.mp3"], ["Grammar-Wrong-Sample.mp3"],], | |
| outputs=['text','text'] | |
| ) | |
| iface.launch(enable_queue=True,show_error=True) |