from speechbrain.inference.ASR import EncoderASR
import gradio as gr

model = EncoderASR.from_hparams("speechbrain/asr-wav2vec2-dvoice-wolof")

def transcribe(audio):
    return model.transcribe_file(audio.name)

demo = gr.Interface(
    fn=transcribe,
    inputs=gr.Audio(sources=["microphone", "upload"], label="Audio en wolof"),
    outputs=gr.Textbox(label="Transcription alphabet latin"),
    title="Transcription audio en wolof latin by PSW",
    description="Ce modèle transcrit un fichier audio en wolof en texte en utilisant l'alphabet latin."
)

demo.launch()