Update app.py

app.py

@@ -82,7 +82,7 @@ def run_interpretation(progress, global_state, raw_interpretation_prompt, max_ne
     # generate the interpretations
     generated = interpretation_prompt.generate(model, {0: interpreted_vectors}, k=3, **generation_kwargs)
     generation_texts = tokenizer.batch_decode(generated)
-    return [gr.TextBox(text, visible=True, container=False) for text in generation_texts]
+    return [gr.Textbox(text, visible=True, container=False) for text in generation_texts]
 
 
 ## main
@@ -183,7 +183,7 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
             for i in range(MAX_PROMPT_TOKENS):
                 btn = gr.Button('', visible=False, elem_classes=['token_btn'])
                 tokens_container.append(btn)
-            interpretation_bubbles = [gr.TextBox('', container=False, visible=False, elem_classes=['bubble'])
+            interpretation_bubbles = [gr.Textbox('', container=False, visible=False, elem_classes=['bubble'])
                                      for i in range(model.config.num_hidden_layers)]
             
         for i, btn in enumerate(tokens_container):