Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
pipe = pipeline(model="Yilin98/whisper-small-hi") # change to "your-username/the-name-you-picked" | |
def transcribe(audio=None, file=None, youtube=None): | |
if (audio is None) and (file is None) and (youtube is None): | |
return "No audio provided!" | |
elif audio is not None: | |
input=audio | |
elif file is not None: | |
input=file | |
elif youtube is not None: | |
yt=pt.YouTube("https://www.youtube.com/watch?v=4KI9BBW_aP8") | |
input=yt.streams.filter(only_audio=True)[0] | |
text = pipe(input)["text"] | |
return text | |
iface = gr.Interface( | |
fn=transcribe, | |
inputs=[ | |
gr.Audio(source="microphone", type="filepath", interactive=True), | |
gr.Audio(source="upload", type="filepath", interactive=True), | |
gr.Text(label="URL (YouTube, etc.)")], | |
outputs="text", | |
title="Whisper Small Swedish", | |
description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model.", | |
) | |
iface.launch() |