Update app.py

app.py

@@ -10,8 +10,6 @@ from transformers.activations import get_activation
 from transformers import AutoTokenizer
 
 
-st.title('KoGPT2 Demo')
-
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 from transformers import AutoTokenizer, AutoModelForMaskedLM
@@ -38,18 +36,4 @@ 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]
-        a_list.append(words)
-      length = len(a_list)
-      middle_index = length//2
-      first_half = a_list[:middle_index]
-      second_half = a_list[middle_index:]
-        #st.write(words)
-        #col1= st.columns(1)
-        #with col1:
-          #st.write(words)
-      st.write(first_half)
-      st.write(second_half)
-      st.write(pd.DataFrame({
-       'first column': first_half,
-       'second column': second_half,
-       }))
+        st.write(words)