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)
|