Update app.py

app.py

@@ -1,9 +1,14 @@
 import streamlit as st
 from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
+from transformers import AutoModelForMaskedLM, AutoTokenizer
 
-text = st.text_area('Enter the text:')
 model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
+model = AutoModelForMaskedLM.from_pretrained("alabnii/jmedroberta-base-sentencepiece")
+modek.eval()
 tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_large="en_XX")
+tokenizer = AutoTokenizer.from_pretrained("alabnii/jmedroberta-base-sentencepiece")
+
+text = st.text_area('Enter the text:')
 
 if text:
     model_inputs = tokenizer(text, return_tensors="pt")