Update app.py

app.py

@@ -44,7 +44,6 @@ def initialize_gpu():
 
 def reset_model(model_name, *extra_components, with_extra_components=True): 
     # extract model info
-    print('HELLO!!')       
     
     model_args = deepcopy(model_info[model_name])
     model_path = model_args.pop('model_path')
@@ -63,8 +62,9 @@ def reset_model(model_name, *extra_components, with_extra_components=True):
     gc.collect()
     if with_extra_components:
         for i in range(len(extra_components)):
-            print('HELLO!!')       
-        return [welcome_message.format(model_name=model_name), *extra_components]
+        return ([welcome_message.format(model_name=model_name)] 
+                + [gr.Textbox(visible=False) for _ in range(len(interpretation_bubbles))] 
+                + extra_components)
     
 
 def get_hidden_states(raw_original_prompt):
@@ -220,8 +220,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
                               [progress_dummy, *tokens_container, *interpretation_bubbles])
     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, *interpretation_bubbles]
-    model_chooser.change(reset_model, [model_chooser, *extra_components], [welcome_model, *extra_components])
+    extra_components = [interpretation_prompt, original_prompt_raw, original_prompt_btn]
+    model_chooser.change(reset_model, [model_chooser, *extra_components], [welcome_model, *interpretation_bubbles, *extra_components])
     
     demo.launch()