Tmeena commited on
Commit
f195f8d
1 Parent(s): 10db1f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
9
  os.environ['HF_HOME'] = '/app/cache'
10
 
11
  # Load the Whisper model with a specified cache directory
12
- whisper_model = whisper.load_model("turbo", download_root="/app/cache")
13
 
14
  # Load the translation model and tokenizer
15
  tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M", cache_dir="/app/cache")
 
9
  os.environ['HF_HOME'] = '/app/cache'
10
 
11
  # Load the Whisper model with a specified cache directory
12
+ whisper_model = whisper.load_model("small", download_root="/app/cache")
13
 
14
  # Load the translation model and tokenizer
15
  tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M", cache_dir="/app/cache")