Update app.py

app.py

@@ -35,4 +35,5 @@ with st.form(key='my_form'):
         mask_hidden_state = last_hidden_state[mask_index]
         idx = torch.topk(mask_hidden_state, k=100, dim=0)[1]
         words = [tokenizer.decode(i.item()).strip() for i in idx]
-        st.text_area(label = 'Infill:', value=words)
+        with st.expander('Expand'):
+          st.text_area(label = 'Infill:', value=words)