working version, youtube not working yet

app.py

@@ -1,34 +1,30 @@
 from transformers import pipeline
 import gradio as gr
-from youtube_dl import YoutubeDL
-
-
-def transcribe_audio(input_type, audio_path):
-    if input_type == "Audio":
-        model = pipeline(model="SofiaK/checkpoints")
-        return model(audio_path)["text"]
+from pytube import YouTube
+import youtube_dl
+import io
+import tempfile
+import os
 
+model = pipeline(model="SofiaK/checkpoints")
 
 with gr.Blocks() as demo:
-    radio = gr.Radio(
-        choices=["Audio", "Youtube"], label="Choose your input type", value="Audio"
-    )
-    audio_input = gr.Audio(
-        sources=["upload", "microphone"],
-        type="filepath",
-        label="Upload Audio, or speak in the microphone",
-        visible=True,
-        interactive=True,
-    )
-    youtube_input = gr.Textbox(
-        value="https://www.youtube.com/",
-        label="Youtube link",
-        visible=False,
-        interactive=True,
-    )
-    btn = gr.Button(
-        "Transcript",
-    )
+    with gr.Row():
+        with gr.Column():
+            radio = gr.Radio(
+                choices=["Audio", "Youtube"],
+                label="Choose your input type: an audio or a youtube link",
+                value="Audio",
+            )
+            audio_input = gr.Audio(
+                sources=["upload", "microphone"], type="filepath", visible=True
+            )
+            youtube_input = gr.Textbox(
+                value="https://www.youtube.com/", label="Youtube Link", visible=False
+            )
+            btn = gr.Button("Submit")
+        with gr.Column():
+            output = gr.Text(label="Model Output")
 
     def make_visible(val):
         audio_visible = val == "Audio"
@@ -39,7 +35,12 @@ with gr.Blocks() as demo:
 
     radio.change(make_visible, inputs=radio, outputs=[audio_input, youtube_input])
 
-    output = gr.Text(label="Model output")
-    btn.click(fn=transcribe_audio, inputs=[radio, audio_input], outputs=output)
+    btn.click(
+        fn=lambda audio_input, youtube_input, radio: model(audio_input)["text"]
+        if radio == "Audio"
+        else "youtube output- not supported yet",
+        inputs=[audio_input, youtube_input, radio],
+        outputs=output,
+    )
 
 demo.launch()