Update app.py
Browse files
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("
|
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")
|