Update app.py

app.py

@@ -38,7 +38,7 @@ with st.expander('BigSalmon/FormalRobertaa'):
         words = [tokenizer.decode(i.item()).strip() for i in idx]
         st.text_area(label = 'Infill:', value=words)
  
-with st.expander('roberta-large'):
+with st.expander('roberta-base result'):
   token_ids = tokenizer.encode(prompt, return_tensors='pt')
   token_ids_tk = tokenizer.tokenize(prompt, return_tensors='pt')
   masked_position = (token_ids.squeeze() == tokenizer.mask_token_id).nonzero()