Update app.py

app.py

@@ -1,5 +1,7 @@
 from transformers import pipeline
 import gradio as gr
+from pytube import YouTube
+import os
 
 pipe = pipeline(model="afroanton/whisper-small-sv-SE")  # change to "your-username/the-name-you-picked"
 
@@ -7,9 +9,19 @@ def transcribe(audio):
     text = pipe(audio)["text"]
     return text
 
+def transcribeUrl(url):
+    yt = YouTube(str(url))
+    audio = yt.streams.filter(only_audio=True).first()
+    text = pipe(audio)["text"]
+    return text
+    
 video_demo = gr.Interface.load(
     fn=transcribe, 
-    inputs="text", 
+    inputs=gr.Textbox(
+            label="insert URL",
+            lines=3,
+            value="The fast brown fox jumps over lazy dogs.",
+        ), 
     outputs="text",
     title="Whisper Small Swedish",
     description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model.",