Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import torch
 
 model_checkpoint = "japanese-denim/mbart-50-finetuned-eng-to-naga"
-# model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
-# tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
+tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
 
 src_lang = 'en_XX'
@@ -13,7 +13,8 @@ tgt_lang = "ng_XX"
 
 def translate(text):
     translation_pipeline = pipeline("translation",
-                                    model=model_checkpoint,
+                                    model=model,
+                                    tokenizer = tokenizer
                                     src_lang=src_lang,
                                     tgt_lang=tgt_lang)