Update app.py

app.py

@@ -157,17 +157,12 @@ for i in range(MAX_PROMPT_TOKENS):
     tokens_container.append(btn)
 
 with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
-    
     with gr.Row():
         with gr.Column(scale=5):
             gr.Markdown('# 😎 Self-Interpreting Models')
 
             gr.Markdown('<b style="color: #8B0000;">Model outputs are not filtered and might include undesired language!</b>')
             
-            # gr.Markdown(
-            #     '**👾 This space is a simple introduction to the emerging trend of models interpreting their OWN hidden states in free form natural language!!👾**',
-            #     # elem_classes=['explanation_accordion']
-            # )
             gr.Markdown(
             '''
                 **👾 This space is a simple introduction to the emerging trend of models interpreting their OWN hidden states in free form natural language!!👾**
@@ -176,9 +171,6 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
                 We will follow the SelfIE implementation in this space for concreteness. Patchscopes are so general that they encompass many other interpretation techniques too!!! 
             ''', line_breaks=True)
             
-            # gr.Markdown('**👾 The idea is really simple: models are able to understand their own hidden states by nature! 👾**',
-            #               # elem_classes=['explanation_accordion']
-            #             )  
             gr.Markdown(
             '''
             **👾 The idea is really simple: models are able to understand their own hidden states by nature! 👾**
@@ -193,7 +185,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     with gr.Group():
         model_chooser = gr.Radio(choices=list(model_info.keys()), value=model_name)
     
-    with gr.Blocks() as demo_blocks:
+    with gr.Group() as demo_blocks:
         gr.Markdown('## Choose Your Interpretation Prompt')
         with gr.Group('Interpretation'):
             interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
@@ -221,7 +213,6 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
             for btn in tokens_container:
                 btn.render()
     
-        
         with gr.Accordion(open=False, label='Generation Settings'):
             with gr.Row():
                 num_tokens = gr.Slider(1, 100, step=1, value=20, label='Max. # of Tokens')