Update app.py

app.py

@@ -3,12 +3,27 @@ import gradio as gr
 
 pipe = pipeline(model="Akseluhr/whisper-small-sv-SE-auhr-v2")
 
-
-def transcribe(rec=None, file=None):
+def get_audio(url):
+  yt = YouTube(url)
+  video = yt.streams.filter(only_audio=True).first()
+  out_file=video.download(output_path=".")
+  base, ext = os.path.splitext(out_file)
+  new_file = base+'.mp3'
+  os.rename(out_file, new_file)
+  a = new_file
+  return a
+
+def get_text(url):
+  result = model.transcribe(get_audio(url))
+  return result['text']
+
+def transcribe(rec=None, file=None, URL=None):
     if rec is not None:
         audio = rec
     elif file is not None:
         audio = file
+    elif url is not None:
+        audio = get_audio(URL)
     else:
         return "Provide a recording or a file."
 
@@ -21,6 +36,7 @@ iface = gr.Interface(
     inputs=[
         gr.Audio(source="microphone", type="filepath", optional=True),
         gr.Audio(source="upload", type="filepath", optional=True),
+        gr.Textbox(placeholder='Enter the Youtube video URL', label='URL', optional=True),
     ],
     outputs="text",
     title="Whisper Small Swedish",