import gradio as gr
from transformers import pipeline

pipe = pipeline("automatic-speech-recognition", model="cportoca/CS224S_Quechua_Project_Bilingual")

def transcribe(audio):
    text = pipe(audio)["text"]
    return text

interface = gr.Interface(
    fn=transcribe,
    inputs=gr.Audio(sources="microphone", type="filepath"),
    outputs="text",
    title="Prototipo de sistema de transcripción de voz en Quechua",
    description="Demostración de transcripción de voz a texto en Quechua utilizando u modelo Wav2Vec2-Bert afinado. Observación: El sistema demora un par de segundos en producir un resultado."
)

interface.launch()