gradio app

app.py

@@ -0,0 +1,24 @@
+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 the URL of the Youtube video clip here (without prefixes like http://):"),
+	outputs="text",
+	title="Whisper Small SE",
+	description="Video Swedish Transcriptior",
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+pytube