Update app.py

app.py

@@ -62,8 +62,8 @@ def reset_model(model_name, *extra_components, with_extra_components=True):
     gc.collect()
     if with_extra_components:
         return ([welcome_message.format(model_name=model_name)] 
-                + [gr.Textbox(visible=False) for _ in range(len(interpretation_bubbles))] 
-                + [gr.Button(visible=False) for _ in range(len(tokens_container))]
+                + [gr.Textbox('', visible=False) for _ in range(len(interpretation_bubbles))] 
+                + [gr.Button('', visible=False) for _ in range(len(tokens_container))]
                 + [*extra_components])
     
 
@@ -113,7 +113,7 @@ def run_interpretation(raw_interpretation_prompt, max_new_tokens, do_sample,
     generation_texts = global_state.tokenizer.batch_decode(generated)
     progress_dummy_output = ''
     bubble_outputs = [gr.Textbox(text.replace('\n', ' '), visible=True, container=False, label=f'Layer {i}') for text in generation_texts]
-    bubble_outputs += [gr.Textbox(visible=False) for _ in range(MAX_NUM_LAYERS - len(bubble_outputs))]
+    bubble_outputs += [gr.Textbox('', visible=False) for _ in range(MAX_NUM_LAYERS - len(bubble_outputs))]
     return [progress_dummy_output, *bubble_outputs]
 
 
@@ -221,6 +221,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     original_prompt_raw.change(lambda: [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)], [], tokens_container)
 
     extra_components = [interpretation_prompt, original_prompt_raw, original_prompt_btn]
-    model_chooser.change(reset_model, [model_chooser, *extra_components], [welcome_model, *interpretation_bubbles, *tokens_container, *extra_components])
+    model_chooser.change(reset_model, [model_chooser, *extra_components], 
+                         [welcome_model, *interpretation_bubbles, *tokens_container, *extra_components])
     
     demo.launch()