updates small100 for each language
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def load_model(model_id: str, target_lang: str):
|
|
58 |
- alirezamsh/small100 via AutoTokenizer & AutoModelForSeq2SeqLM
|
59 |
- all others via MarianTokenizer & MarianMTModel
|
60 |
"""
|
61 |
-
if model_id not in MODEL_CACHE:
|
62 |
if model_id.startswith("facebook/mbart"):
|
63 |
tokenizer = MBart50TokenizerFast.from_pretrained(model_id)
|
64 |
model = MBartForConditionalGeneration.from_pretrained(model_id)
|
|
|
58 |
- alirezamsh/small100 via AutoTokenizer & AutoModelForSeq2SeqLM
|
59 |
- all others via MarianTokenizer & MarianMTModel
|
60 |
"""
|
61 |
+
if model_id not in MODEL_CACHE or model_id == "alirezamsh/small100":
|
62 |
if model_id.startswith("facebook/mbart"):
|
63 |
tokenizer = MBart50TokenizerFast.from_pretrained(model_id)
|
64 |
model = MBartForConditionalGeneration.from_pretrained(model_id)
|