Update app.py

app.py

@@ -111,6 +111,7 @@ json_output = gr.JSON()
 css = '''
 
 .bubble {
+  border: none
   border-radius: 10px;
   padding: 10px;
   margin-top: 15px;
@@ -186,14 +187,9 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
         with gr.Column(scale=1):    
             gr.Markdown('<span style="font-size:180px;">🤔</span>')
 
-    # with gr.Group():
-    #     with gr.Row():
-    #         for txt in model_info.keys():
-    #             btn = gr.Button(txt)
-    #             model_btns.append(btn)
-    #         for btn in model_btns:
-    #             btn.click(reset_new_model, [global_state])
-    
+    with gr.Group('Interpretation'):
+        interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
+
     with gr.Blocks():
         gr.Markdown('''
         Here are some examples of prompts we can analyze their internal representations
@@ -219,9 +215,6 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
     
     progress_dummy = gr.Markdown('', elem_id='progress_dummy')
 
-    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)]
@@ -239,6 +232,14 @@ 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')
 
+    # with gr.Group():
+    #     with gr.Row():
+    #         for txt in model_info.keys():
+    #             btn = gr.Button(txt)
+    #             model_btns.append(btn)
+    #         for btn in model_btns:
+    #             btn.click(reset_new_model, [global_state])
+
     # event listeners
     for i, btn in enumerate(tokens_container):
         btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt,