Update app.py

app.py

@@ -2,17 +2,16 @@ import gradio as gr
 import whisper
 import os
 
-model_size = list(whisper._MODELS.keys())
+model = whisper.load_model("base")
 
 # main processing
-def main(URL, model_size):
+def main(URL):
     video_path = youtube_dl(URL)
-    text = transcript(video_path, model_size)
+    text = transcript(video_path)
     return text
 
 # Transcribe text using Whisper
-def transcript(video_path, model_size):
-    model = whisper.load_model(model_size)
+def transcript(video_path):
     result = model.transcribe(video_path)
     return result["text"]
 
@@ -40,9 +39,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 url = gr.Textbox(placeholder = 'Youtube video URL', label = 'URL')
 
-            with gr.Row():
-                model_size = gr.Dropdown(choices = model_size, value = 'tiny', label = "Model")
-
             with gr.Row():
                 transcribe_btn = gr.Button('Transcribe')
 
@@ -50,7 +46,7 @@ with gr.Blocks() as demo:
             outputs = gr.Textbox(placeholder = 'Transcription of the video', label = 'Transcription')
 
     transcribe_btn.click(fn = main,
-                        inputs = [url, model_size],
+                        inputs = url,
                         outputs = outputs
                         )