Update app.py

app.py

@@ -159,7 +159,9 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
 
         # with gr.Column(scale=1):    
         #     gr.Markdown('<span style="font-size:180px;">🤔</span>')
-
+    gr.Markdown('''
+    ## Choose Your Interpretation Prompt 
+    ''')
     with gr.Group('Interpretation'):
         interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
         gr.Examples([[p] for p in suggested_interpretation_prompts], [interpretation_prompt], cache_examples=False)
@@ -176,7 +178,9 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     #         dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
     #         dataset = [[row[info['text_col']]] for row in dataset]
     #         gr.Examples(dataset, [original_prompt_raw])
-                
+    gr.Markdown('''
+    ## The Prompt to Analyze 
+    ''')        
     with gr.Group():
         original_prompt_raw.render()
         original_prompt_btn = gr.Button('Compute', variant='primary')
@@ -189,9 +193,11 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     use_gpu = False # gr.Checkbox(value=False, label='Use GPU')
     progress_dummy = gr.Markdown('', elem_id='progress_dummy')
 
+    
     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():