Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import torch
 
-model_checkpoint = "japanese-denim/mbart-50-finetuned-naga-to-eng"
+model_checkpoint = "japanese-denim/m2m-finetuned-naga-to-eng"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
@@ -10,8 +10,8 @@ def translate(text):
     translation_pipeline = pipeline("translation",
                                     model=model,
                                     tokenizer = tokenizer,
-                                    src_lang='ng_XX',
-                                    tgt_lang='en_XX')
+                                    src_lang='en',
+                                    tgt_lang='en')
 
     result = translation_pipeline(text)
     return result[0]['translation_text']