Update app.py

app.py

@@ -82,7 +82,7 @@ def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens,
     # generate the interpretations
     generated = interpretation_prompt.generate(model, {0: interpreted_vectors}, k=3, **generation_kwargs)
     generation_texts = tokenizer.batch_decode(generated)
-    return [gr.Text(text, visible=True) for text in generation_texts]
+    return [gr.Text(text, visible=True, container=False) for text in generation_texts]
 
 
 ## main
@@ -136,7 +136,7 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
     with gr.Row():
         with gr.Column(scale=5):
             gr.Markdown('''
-                # 😎 Self-Interpreting Models 😎
+                # 😎 Self-Interpreting Models
                 
                 👾 **This space is a simple introduction to the emerging trend of models interpreting their _own hidden states_ in free form natural language**!! 👾
                 This idea was explored in the paper **Patchscopes** ([Ghandeharioun et al., 2024](https://arxiv.org/abs/2401.06102)) and was later investigated further in **SelfIE** ([Chen et al., 2024](https://arxiv.org/abs/2403.10949)). 
@@ -179,7 +179,7 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
                 tokens_container.append(btn)
         
         for i in range(model.config.num_hidden_layers):
-            interpretation_bubbles.append(gr.Text('', visible=False, elrm_classes=['bubble']))
+            interpretation_bubbles.append(gr.Text('', visible=False, elem_classes=['bubble']))
 
         for i, btn in enumerate(tokens_container):
             btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, num_tokens, do_sample, temperature,