Spaces:
Runtime error
Runtime error
File size: 533 Bytes
39ec15a c57f3e0 39ec15a ece4569 39ec15a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from transformers import pipeline
from pytube import YouTube
import gradio as gr
pipe = pipeline(model="id2223lab1/whisper-small")
def transcribe(url):
audio = YouTube(url).streams.filter(file_extension='mp4', only_audio=True).first().download()
text = pipe(audio, batch_size=512, truncation=True)["text"]
return text
iface = gr.Interface(
fn=transcribe,
inputs=gr.Textbox(label="Enter a YouTube URL:"),
outputs="text",
title="Whisper Small SE",
description="Transcribe swedish videos from YouTube",
)
iface.launch()
|