seanp03 commited on
Commit
1d177ae
·
verified ·
1 Parent(s): 74b0636

updates small100 for each language

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)