youtube works

app.py

@@ -1,12 +1,16 @@
 from transformers import pipeline
 import gradio as gr
 from pytube import YouTube
-import youtube_dl
-import io
-import tempfile
-import os
 
-model = pipeline(model="SofiaK/checkpoints")
+model = pipeline(model="SofiaK/dataset")
+
+
+def youtube_to_text(youtube_url):
+    video = YouTube(youtube_url).streams.filter(only_audio=True).all()
+    audio = video[0].download()
+    text = model(audio)["text"]
+    return text
+
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -38,7 +42,7 @@ with gr.Blocks() as demo:
     btn.click(
         fn=lambda audio_input, youtube_input, radio: model(audio_input)["text"]
         if radio == "Audio"
-        else "youtube output- not supported yet",
+        else youtube_to_text(youtube_input),
         inputs=[audio_input, youtube_input, radio],
         outputs=output,
     )