Update app.py

app.py

@@ -14,11 +14,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 tokenizer = AutoTokenizer.from_pretrained("roberta-large")
-model = AutoModelForMaskedLM.from_pretrained("BigSalmon/FormalRobertaa")
+model = AutoModelForMaskedLM.from_pretrained("BigSalmon/FormalRobertaaa")
 model2 = AutoModelForMaskedLM.from_pretrained("roberta-base")
 
 
-with st.expander('BigSalmon/FormalRobertaa'):
+with st.expander('BigSalmon/FormalRobertaaa'):
   with st.form(key='my_form'):
     prompt = st.text_area(label='Enter Text. Put <mask> where you want the model to fill in the blank. You can use more than one at a time.')
     submit_button = st.form_submit_button(label='Submit')