Update app.py

app.py

@@ -77,7 +77,7 @@ def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens,
     }
     
     # create an InterpretationPrompt object from raw_interpretation_prompt (after putting it in the right template)
-    interpretation_prompt = interpretation_prompt_template.format(prompt=raw_interpretation_prompt)
+    interpretation_prompt = interpretation_prompt_template.format(prompt=raw_interpretation_prompt, repeat=1)
     interpretation_prompt = InterpretationPrompt(tokenizer, interpretation_prompt)
 
     # generate the interpretations
@@ -235,10 +235,9 @@ 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('Output'):
-        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)]
+    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)]
         
     for i, btn in enumerate(tokens_container):
         btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt,