Update app.py

app.py

@@ -27,12 +27,12 @@
 # # Launch the interface
 # gr.Interface(fn=predict, inputs=gr.inputs.Textbox, outputs=gr.outputs.Textbox).launch(share=False)
 
-from transformers import T5ForConditionalGeneration, MBart50Tokenizer
+from transformers import MBartForConditionalGeneration, MBart50Tokenizer
 import gradio as grad
 
 model_name = "MiVaCod/mbart-neutralization"
 text2text_tkn= MBart50Tokenizer.from_pretrained(model_name)
-mdl = T5ForConditionalGeneration.from_pretrained(model_name)
+mdl = MBartForConditionalGeneration.from_pretrained(model_name)
 
 def text2text_paraphrase(sentence1):
      inp1 = "rte sentence1: "+sentence1