from transformers import pipeline
import gradio as gr


# Load the pipeline with the cache_dir parameter
pipe = pipeline(model="tarteel-ai/whisper-base-ar-quran")

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

iface = gr.Interface(
    fn=transcribe,
    inputs=gr.Audio(source="upload", type="filepath"),
    outputs="text",
)

iface.launch()