richylyq commited on
Commit
b0ae3fd
1 Parent(s): 9e1c3ab
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -27,13 +27,13 @@ import langid
27
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
28
  from easynmt import EasyNMT
29
 
30
- # Initialize nllb-200 models
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")
 
27
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
28
  from easynmt import EasyNMT
29
 
30
+ # # Initialize nllb-200 models
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")