Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import whisper
 from pytube import YouTube
 
-loaded_model = whisper.load_model("base")
-current_size = 'base'
+loaded_model = whisper.load_model("medium")
+current_size = 'medium'
 def inference(link):
   yt = YouTube(link)
   path = yt.streams.filter(only_audio=True)[0].download(filename="audio.mp4")
@@ -35,7 +35,7 @@ with block:
     )
     with gr.Group():
         with gr.Box():
-          sz = gr.Dropdown(label="Model Size", choices=['base','small', 'medium', 'large'], value='base')
+          sz = gr.Dropdown(label="Model Size", choices=['tiny', 'base','small', 'medium', 'large'], value='medium')
           
           link = gr.Textbox(label="YouTube Link")