Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from transformers import MBartForConditionalGeneration, MBart50Tokenizer
+from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
 # model_ru_qm_path = 'TSjB/mbart-large-52-ru-qm-v1'
 # model_qm_ru_path = 'TSjB/mbart-large-52-qm-ru-v1'
 MODEL_PATH = 'TSjB/NLLB-201-600M-QM-V1'
@@ -10,8 +10,8 @@ MODEL_PATH = 'TSjB/NLLB-201-600M-QM-V1'
 #tokenizer_qm_ru = MBart50Tokenizer.from_pretrained(model_qm_ru_path)
 #model_ru_qm = MBartForConditionalGeneration.from_pretrained(model_ru_qm_path)
 #model_qm_ru = MBartForConditionalGeneration.from_pretrained(model_qm_ru_path)
-tokenizer = MBart50Tokenizer.from_pretrained(MODEL_PATH)
-model = MBartForConditionalGeneration.from_pretrained(MODEL_PATH)
+tokenizer = NllbTokenizer.from_pretrained(MODEL_PATH)
+model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_PATH)
 
 # 3. Fix tokenizer
 def fixTokenizer(tokenizer, new_lang='krc_Cyrl'):