Hugging Face's logo

Spaces:
chompionsawelo
/
whisper_transcribe
Runtime error

App Files Community
1
chompionsawelo commited on
Commit
8f3ea36
·
1 Parent(s): 605f620

fix for model size large

Browse files
Files changed (1) hide show
  1. transcribe.py +1 -1
transcribe.py CHANGED
@@ -9,7 +9,7 @@ gc.collect()
9
  torch.cuda.empty_cache()
10
 
11
  model_lang_list = ['en', 'id', None]
12
- model_size = ["tiny", "base", "small", "medium", "large"]
13
 
14
 
15
  def start_transcribe(input_file, lang_choice: int, model_size_choice: int, progress):
 
9
  torch.cuda.empty_cache()
10
 
11
  model_lang_list = ['en', 'id', None]
12
+ model_size = ["tiny", "base", "small", "medium", "large-v2"]
13
 
14
 
15
  def start_transcribe(input_file, lang_choice: int, model_size_choice: int, progress):