Changed the buttons position

app.py

@@ -78,10 +78,10 @@ def return_top_k(sentence, k=10, word=None, rels=False):
     predictions = [tokenizer.decode(tokens, skip_special_tokens=True) for tokens in output_sequences['sequences']]
     probabilities = [round(float(prob), 2) for prob in decoded_probabilities]
 
-    # stripped_sent = [remove_punctuation(word.lower()) for word in sentence.split()]
-    # for pred in predictions:
-    #   if (len(pred) < 2) | (pred in stripped_sent):
-    #     predictions.pop(predictions.index(pred))  
+    stripped_sent = [remove_punctuation(word.lower()) for word in sentence.split()]
+    for pred in predictions:
+      if (len(pred) < 2) | (pred in stripped_sent):
+        predictions.pop(predictions.index(pred))  
      
   return predictions[:10] 
 
@@ -303,8 +303,8 @@ if st.session_state.actions[-1] == 'cue':
     word_count = st.session_state.counters["word_count"]
     target = st.session_state.results["results"][word_count]
     
-    col1, col2, col3, col4, col11, col12 = st.columns(6)
-    col5, col6, col7, col8, col9 = st.columns(5)
+    col1, col2, col3, col4, col5 = st.columns(5)
+  
 
     with col1:
       b1 = st.button("Next letter", key="1")
@@ -320,13 +320,9 @@ if st.session_state.actions[-1] == 'cue':
      # avail_cues = get_available_cues(target)
       #cues_buttons = {cue_type: st.button(cue_type) for cue_type in avail_cues}
 
-    with col5:
-      b5 = st.button("I remembered the word!", key="4", type='primary')
-    with col6:
-      b6 = st.button("Exit", key="5", type='primary')
-
-    with col7:
-      new = st.button('Play again', key=64)
+    b5 = st.button("I remembered the word!", key="4", type='primary')
+    b6 = st.button("Exit", key="5", type='primary')
+    new = st.button('Play again', key=64)
 
     if b1:
       st.session_state.counters["letter_count"] += 1