Update app.py (#2)

- Update app.py (7d9c01d724a370a36baea1f3c376ad18a5eba39a)


Co-authored-by: Zineb Senane <ZinebSN@users.noreply.huggingface.co>

app.py

@@ -3,13 +3,16 @@ 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):
-    if (audio is None) and (file is None):
+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
 
@@ -17,7 +20,8 @@ iface = gr.Interface(
     fn=transcribe, 
     inputs=[
         gr.Audio(source="microphone", type="filepath", interactive=True),
-        gr.Audio(source="upload", 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.",