richylyq commited on
Commit
7239594
1 Parent(s): 86cec43

add back mbart model

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -31,9 +31,9 @@ from easynmt import EasyNMT
31
  # tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
32
  # model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
33
 
34
- # # Initialize mbart50 models
35
- # mbart_m2en_model = EasyNMT("mbart50_m2en")
36
- # mbart_en2m_model = EasyNMT("mbart50_en2m")
37
 
38
  # Initialize m2m_100 models
39
  m2m_model = EasyNMT("m2m_100_1.2B")
 
31
  # tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
32
  # model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
33
 
34
+ # Initialize mbart50 models
35
+ mbart_m2en_model = EasyNMT("mbart50_m2en")
36
+ mbart_en2m_model = EasyNMT("mbart50_en2m")
37
 
38
  # Initialize m2m_100 models
39
  m2m_model = EasyNMT("m2m_100_1.2B")