Update app.py

app.py

@@ -111,7 +111,6 @@ json_output = gr.JSON()
 css = '''
 
 .bubble {
-  border: 2px solid #000;
   border-radius: 10px;
   padding: 10px;
   margin-top: 15px;
@@ -223,6 +222,10 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
     with gr.Group('Interpretation'):
         interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
 
+    interpretation_bubbles = [gr.Textbox('', container=False, visible=False, elem_classes=['bubble', 
+                                                                                           'even_bubble' if i % 2 == 0 else 'odd_bubble'])
+                             for i in range(model.config.num_hidden_layers)]
+    
     with gr.Accordion(open=False, label='Settings'):
         with gr.Row():
             num_tokens = gr.Slider(1, 100, step=1, value=20, label='Max. # of Tokens')
@@ -236,10 +239,7 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
                 top_k = gr.Slider(1, 1000, value=50, step=1, label='top k')
                 top_p = gr.Slider(0., 1., value=0.95, label='top p')
 
-    interpretation_bubbles = [gr.Textbox('', container=False, visible=False, elem_classes=['bubble', 
-                                                                                           'even_bubble' if i % 2 == 0 else 'odd_bubble'])
-                             for i in range(model.config.num_hidden_layers)]
-        
+    # event listeners
     for i, btn in enumerate(tokens_container):
         btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, 
                                                      num_tokens, do_sample, temperature,