Yilin98's picture
Update app.py (#2)
9430119
raw
history blame
1.03 kB
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()