Update app.py

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, col13 = st.columns(7)
-    col5, col6, col7, col8, col9, col10 = st.columns(6)
+    col1, col2, col3, col4, col11, col12 = st.columns(6)
+    col5, col6, col7, col8, col9 = st.columns(5)
 
     with col1:
       b1 = st.button("Next letter", key="1")