Update app.py

app.py

@@ -2,16 +2,17 @@ import gradio as gr
 import whisper
 import os
 
-model = whisper.load_model("tiny")
+model_size = list(whisper._MODELS.keys())
 
 # main processing
-def main(URL):
+def main(URL, model_size):
     video_path = youtube_dl(URL)
-    text = transcript(video_path)
+    text = transcript(video_path, model_size)
     return text
 
 # Transcribe text using Whisper
-def transcript(video_path):
+def transcript(video_path, model_size):
+    model = whisper.load_model(model_size)
     result = model.transcribe(video_path)
     return result["text"]
 
@@ -23,8 +24,28 @@ def youtube_dl(URL):
     video_path = os.path.join(os.path.dirname(__file__), "target.mp4")
     return video_path
 
-demo = gr.Interface(fn = main,
-                    inputs = "text", outputs = "text")
+with gr.Blocks() as demo:
+
+    with gr.Row():
+
+        with gr.Column():
+
+            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')
+
+        with gr.Column():
+            outputs = gr.Textbox(placeholder = 'Transcription of the video', label = 'Transcription')
+
+    transcribe_btn.click(fn = main,
+                        inputs = [url, model_size],
+                        outputs = outputs
+                        )
 
 if __name__ == "__main__":
     demo.launch()