LucFast commited on
Commit
6f76e04
·
1 Parent(s): 2c94bc3

update default model size and placement

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,7 +8,7 @@ class GradioInference():
8
  def __init__(self):
9
  self.sizes = list(whisper._MODELS.keys())
10
  self.langs = ["none"] + sorted(list(whisper.tokenizer.LANGUAGES.values()))
11
- self.current_size = "base"
12
  self.loaded_model = whisper.load_model(self.current_size)
13
  self.yt = None
14
 
@@ -132,7 +132,6 @@ with block:
132
  title = gr.Label(label="Video Title")
133
  with gr.Row().style(equal_height=True):
134
  img = gr.Image(label="Thumbnail")
135
- with gr.Row().style(equal_height=True):
136
  transcript = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
137
  translate = gr.Textbox(label="Translation", placeholder="Translation Output", lines=10)
138
  with gr.Row().style(equal_height=True):
 
8
  def __init__(self):
9
  self.sizes = list(whisper._MODELS.keys())
10
  self.langs = ["none"] + sorted(list(whisper.tokenizer.LANGUAGES.values()))
11
+ self.current_size = "medium"
12
  self.loaded_model = whisper.load_model(self.current_size)
13
  self.yt = None
14
 
 
132
  title = gr.Label(label="Video Title")
133
  with gr.Row().style(equal_height=True):
134
  img = gr.Image(label="Thumbnail")
 
135
  transcript = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
136
  translate = gr.Textbox(label="Translation", placeholder="Translation Output", lines=10)
137
  with gr.Row().style(equal_height=True):