Update app.py

app.py

@@ -35,4 +35,7 @@ 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.write(words)
+        st.write(words)
+        col1= st.columns(1)
+        with col1:
+          st.write(words)