minor instruction update

app.py

@@ -244,9 +244,11 @@ def reset_weights(contextualization_weights):
     contextualization_weights = None
     return contextualization_weights
 
-with gr.Blocks( css = """#sense0slider, #sense1slider, #sense2slider, #sense3slider, #sense4slider, #sense5slider, #sense6slider, #sense7slider,
+with gr.Blocks( theme = gr.themes.Base(), 
+                css = """#sense0slider, #sense1slider, #sense2slider, #sense3slider, #sense4slider, #sense5slider, #sense6slider, #sense7slider,
                         #sense8slider, #sense9slider, #sense1slider0, #sense11slider, #sense12slider, #sense13slider, #sense14slider, #sense15slider 
-                        { height: 200px; width: 200px;  transform: rotate(270deg); }""" ) as demo:
+                        { height: 200px; width: 200px;  transform: rotate(270deg); }""" 
+                ) as demo:
     
     gr.Markdown("""
     ## Backpack Sense Visualization
@@ -259,21 +261,22 @@ with gr.Blocks( css = """#sense0slider, #sense1slider, #sense2slider, #sense3sli
         top_k = gr.State(10)
         with gr.Row():
             with gr.Column(scale=8):
-                input_sentence = gr.Textbox(label="Input Sentence", placeholder='Enter a sentence and click "Predict next word"')
+                input_sentence = gr.Textbox(label="Input Sentence", placeholder='Enter a sentence and click "Predict next word". Then, you can go to the Tokens section, click on a token, and see its contextualization weights.')
             with gr.Column(scale=1):
                 predict = gr.Button(value="Predict next word", variant="primary")
                 reset_weights_button = gr.Button("Reset weights")
-        top_k_words = gr.Dataframe(label="Next Word Predictions (top k)")
-        gr.Markdown("""### **Tokens:** click on a token to see its senses and contextualization weights""")
-        tokens = gr.DataFrame(label="")
+        gr.Markdown("""#### Top-k predicted next word""")
+        top_k_words = gr.Dataframe(interactive=False)
+        gr.Markdown("""### **Token Breakdown:** click on a token below to see its senses and contextualization weights""")
+        tokens = gr.DataFrame()
         with gr.Row():
             with gr.Column(scale=1):
                 selected_token = gr.Textbox(label="Current Selected Token", interactive=False)
             with gr.Column(scale=8):
                 gr.Markdown("""#####
-                Once a token is chosen, you can use the sliders below to change the weights of any senses for that token, \
+                Once a token is chosen, you can **use the sliders below to change the weights of any senses** for that token, \
                     and then click "Predict next word" to see updated next-word predictions. \
-                    You can change the weights of *multiple senses of multiple tokens*; \
+                    You can change the weights of *multiple senses of multiple tokens;* \
                     changes will be preserved until you click "Reset weights".
                 """)
         # sense sliders and top sense words dataframes
@@ -296,21 +299,21 @@ with gr.Blocks( css = """#sense0slider, #sense1slider, #sense2slider, #sense3sli
                 sense7slider= gr.Slider(minimum=0, maximum=1, value=0, step=0.01, label="Sense 7", elem_id="sense7slider", interactive=True)
         with gr.Row():
             with gr.Column(scale=0, min_width=120):
-                sense0words = gr.DataFrame()
+                sense0words = gr.DataFrame(headers = ["Sense 0"])
             with gr.Column(scale=0, min_width=120):
-                sense1words = gr.DataFrame()
+                sense1words = gr.DataFrame(headers = ["Sense 1"])
             with gr.Column(scale=0, min_width=120):
-                sense2words = gr.DataFrame()
+                sense2words = gr.DataFrame(headers = ["Sense 2"])
             with gr.Column(scale=0, min_width=120):
-                sense3words = gr.DataFrame()
+                sense3words = gr.DataFrame(headers = ["Sense 3"])
             with gr.Column(scale=0, min_width=120):
-                sense4words = gr.DataFrame()
+                sense4words = gr.DataFrame(headers = ["Sense 4"])
             with gr.Column(scale=0, min_width=120):
-                sense5words = gr.DataFrame()
+                sense5words = gr.DataFrame(headers = ["Sense 5"])
             with gr.Column(scale=0, min_width=120):
-                sense6words = gr.DataFrame()
+                sense6words = gr.DataFrame(headers = ["Sense 6"])
             with gr.Column(scale=0, min_width=120):
-                sense7words = gr.DataFrame()
+                sense7words = gr.DataFrame(headers = ["Sense 7"])
         with gr.Row():
             with gr.Column(scale=0, min_width=120):
                 sense8slider= gr.Slider(minimum=0, maximum=1, value=0, step=0.01, label="Sense 8", elem_id="sense8slider", interactive=True)
@@ -330,21 +333,21 @@ with gr.Blocks( css = """#sense0slider, #sense1slider, #sense2slider, #sense3sli
                 sense15slider= gr.Slider(minimum=0, maximum=1, value=0, step=0.01, label="Sense 15", elem_id="sense15slider", interactive=True)
         with gr.Row():
             with gr.Column(scale=0, min_width=120):
-                sense8words = gr.DataFrame()
+                sense8words = gr.DataFrame(headers = ["Sense 8"])
             with gr.Column(scale=0, min_width=120):
-                sense9words = gr.DataFrame()
+                sense9words = gr.DataFrame(headers = ["Sense 9"])
             with gr.Column(scale=0, min_width=120):
-                sense10words = gr.DataFrame()
+                sense10words = gr.DataFrame(headers = ["Sense 10"])
             with gr.Column(scale=0, min_width=120):
-                sense11words = gr.DataFrame()
+                sense11words = gr.DataFrame(headers = ["Sense 11"])
             with gr.Column(scale=0, min_width=120):
-                sense12words = gr.DataFrame()
+                sense12words = gr.DataFrame(headers = ["Sense 12"])
             with gr.Column(scale=0, min_width=120):
-                sense13words = gr.DataFrame()
+                sense13words = gr.DataFrame(headers = ["Sense 13"])
             with gr.Column(scale=0, min_width=120):
-                sense14words = gr.DataFrame()
+                sense14words = gr.DataFrame(headers = ["Sense 14"])
             with gr.Column(scale=0, min_width=120):
-                sense15words = gr.DataFrame()
+                sense15words = gr.DataFrame(headers = ["Sense 15"])
         
         # gr.Examples(
         #     examples=[["Messi plays for", top_k, None]],