test commit

app.py

@@ -61,7 +61,8 @@ def run_chains(tokenizer,model,mask_id,input_text,num_steps):
     for step_id in range(num_steps):
         chain_progress.progress((step_id+1)/num_steps)
         pos = torch.randint(seq_len-2,size=(1,)).item()+1
-        data_list.append([step_id,' '.join([tokenizer.decode([token]) for token in sentence[0]]),pos])
+        #data_list.append([step_id,' '.join([tokenizer.decode([token]) for token in sentence[0]]),pos])
+        data_list.append([step_id,tokenizer.decode([token for token in sentence[0]]),pos])
         probs = mask_prob(model,mask_id,sentence,pos)
         sentence,_ = sample_words(probs,pos,sentence)
     return pd.DataFrame(data=data_list,columns=['step','sentence','next_sample_loc'])
@@ -93,7 +94,6 @@ def update_sent_id(increment_value=0):
 def initialize_sent_id():
     st.session_state.sent_id = st.session_state.sent_id_from_slider
 
-
 if __name__=='__main__':
     # Config
     max_width = 1500
@@ -124,6 +124,7 @@ if __name__=='__main__':
 
     # Title
     st.header("Demo: Probing BERT's priors with serial reproduction chains")
+    # Descriptions
     with st.expander("Expand to read the descriptions"):
         st.text("Let's explore sentences in the serial reproduction chains generated by BERT!")
         st.text("First, please choose the samples from the two pre-generated chains,")