import gradio as gr
from transformers import pipeline
inputs=gr.Audio(type="filepath")
pipe = pipeline(model="models/TruongScotl/stvietnamese2")
title = "Speech to Text Translation"
description = """
Open your eyes Morty, that translator is mind-blowing. Drop the damn file, Morty!
"""
def translate(audio):
text = pipe(audio)["text"]
return text
translate = gr.Interface(
fn=translate,
inputs=inputs,
outputs="text",
title=title,
description=description
)
translate.launch()