included more instructions

app.py

@@ -94,14 +94,14 @@ def plot_fig(df,sent_id,xlims,ylims,color_list):
     ax.axis('off')
     ax.set_title(df.cleaned_sentence.to_list()[sent_id])
     #fig.savefig(f'figures/{sent_id}.png')
-    #plt.clf()
-    #plt.close()
     buf = io.BytesIO()
     fig.savefig(buf, format="png", dpi=200)
     buf.seek(0)
     img_arr = np.frombuffer(buf.getvalue(), dtype=np.uint8)
     buf.close()
     img = cv2.imdecode(img_arr, 1)
+    plt.clf()
+    plt.close()
     return img
 
 def pre_render_images(df,input_sent_id):
@@ -153,15 +153,17 @@ if __name__=='__main__':
 
     # Title
     st.header("Demo: Probing BERT's priors with serial reproduction chains")
-
+    st.text("Explore sentences in the serial reproduction chains generated by BERT!")
+    st.text("Visit different positions in the chain using the widgets on the left.")
+    st.text("Check 'Show candidates' to see what words are proposed when each word is masked out.")
     # Load BERT
     tokenizer,model = load_model('bert-base-uncased')
     mask_id = tokenizer.encode("[MASK]")[1:-1][0]
 
     # First step: load the dataframe containing sentences
-    input_type = st.sidebar.radio(label='1. Choose the input type',options=('Use one of our example sentences','Use your own initial sentence'))
+    input_type = st.sidebar.radio(label='1. Choose the input type',options=('Use one of the example sentences','Use your own initial sentence'))
 
-    if input_type=='Use one of our example sentences':
+    if input_type=='Use one of the example sentences':
         sentence = st.sidebar.selectbox("Select the inital sentence",
                                 ('About 170 campers attend the camps each week.',
                                 'She grew up with three brothers and ten sisters.'))
@@ -173,7 +175,7 @@ if __name__=='__main__':
         st.session_state.df = load_data(sentence_num)
 
     else:
-        sentence = st.sidebar.text_input('Type down your own sentence here',on_change=clear_df)
+        sentence = st.sidebar.text_input('Type down your own sentence here.',on_change=clear_df)
         num_steps = st.sidebar.number_input(label='How many steps do you want to run?',value=1000)
         if st.sidebar.button('Run chains'):
             chain = run_chains(tokenizer,model,mask_id,sentence,num_steps=num_steps)
@@ -182,15 +184,15 @@ if __name__=='__main__':
 
     if 'df' in st.session_state:
         df = st.session_state.df
-        sent_id = st.sidebar.slider(label='2. Select the position in a chain to start exploring',
+        sent_id = st.sidebar.slider(label='2. Select a position in the chain to start exploring',
                                     min_value=0,max_value=len(df)-1,value=0)
 
-        if input_type=='Use one of our example sentences':
+        if input_type=='Use one of the example sentences':
             explore_type = st.sidebar.radio('3. Choose the way to explore',options=['In fixed increments','Click through each step','Autoplay'])
         else:
             explore_type = st.sidebar.radio('3. Choose the way to explore',options=['In fixed increments','Click through each step'])
         if explore_type=='Autoplay':
-            if st.button('Create the video (this may take a few minutes)'):
+            #if st.button('Create the video (this may take a few minutes)'):
                 #st.write('Creating the video...')
                 #x_tsne, y_tsne = df.x_tsne, df.y_tsne
                 #xscale_unit = (max(x_tsne)-min(x_tsne))/10
@@ -211,10 +213,10 @@ if __name__=='__main__':
                     #img = cv2.imread(f'figures/{sent_id}.png')
                 #    out.write(img)
                 #out.release()
-                cols = st.columns([1,2,1])
-                with cols[1]:
-                    with open(f'sampling_video_{sentence_num}.mp4', 'rb') as f:
-                        st.video(f)
+            cols = st.columns([1,2,1])
+            with cols[1]:
+                with open(f'sampling_video_{sentence_num}.mp4', 'rb') as f:
+                    st.video(f)
         else:
             if explore_type=='In fixed increments':
                 button_labels = ['-500','-100','-10','-1','0','+1','+10','+100','+500']