Update app.py

app.py

@@ -11,6 +11,13 @@ def transcribe_and_translate(audio):
     translation = translation_pipe(transcription)[0]['translation_text']
     return transcription, translation
 
+
+def transcribe_and_translate_yt(link):
+    yt = YouTube(video_link)
+    audio = yt.streams.filter(only_audio=True).first().download()
+
+    return transcribe_and_translate(audio)
+
 with gr.Blocks() as demo_realtime:
     with gr.Tab("Real-time Swedish to English Transcription and Translation"):
         gr.Interface(
@@ -18,17 +25,17 @@ with gr.Blocks() as demo_realtime:
             inputs=gr.Audio(source="microphone", type="filepath"),
             outputs=[gr.Textbox(), gr.Textbox()],
             title="Whisper Small Swedish to English Translator",
-            description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model, and Swedish to English translation using a small T5 model",
+            description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model, and Swedish to English translation",
         )
         
 with gr.Blocks() as demo_youtube:
     with gr.Tab("Youtube Video Transcription and Translation"):
        gr.Interface(
-            fn=transcribe_and_translate,
-            inputs=gr.Audio(source="microphone", type="filepath"),
+            fn=transcribe_and_translate_youtube,
+            inputs=gr.Textbox("text", label="Enter YouTube Video Link"),
             outputs=[gr.Textbox(), gr.Textbox()],
-            title="Whisper Small Swedish to English Translator",
-            description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model, and Swedish to English translation using a small T5 model",
+            title="Whisper Small Swedish to English Translator - Youtube Video",
+            description="Paste the link of a YouTube Video in Swedish to transcribe and translate it",
         )
 
 demo_realtime.launch(debug=True)