Maxkillor commited on
Commit
73b0620
·
verified ·
1 Parent(s): 70a6679

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  import tempfile
8
 
9
  # Load the Whisper models once at startup
10
- model_sizes = ['tiny', 'base', 'small']
11
  models = {size: whisper.load_model(size) for size in model_sizes}
12
 
13
  # Task options
 
7
  import tempfile
8
 
9
  # Load the Whisper models once at startup
10
+ model_sizes = ['tiny', 'base', 'small', 'medium', 'large']
11
  models = {size: whisper.load_model(size) for size in model_sizes}
12
 
13
  # Task options