Update app.py

app.py

@@ -16,6 +16,7 @@ from configs import model_info, dataset_info
 
 MAX_PROMPT_TOKENS = 60
 MAX_NUM_LAYERS = 50
+welcome_message = '**You are now running {model_name}!!** 🥳🥳🥳'
 
 @dataclass
 class GlobalState:
@@ -63,7 +64,7 @@ def reset_model(model_name, *extra_components, with_extra_components=True):
     if with_extra_components:
         for i in range(len(extra_components)):
             print('HELLO!!')       
-        return extra_components
+        return [welcome_message.format(model_name=model_name), *extra_components]
     
 
 def get_hidden_states(raw_original_prompt):
@@ -157,7 +158,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
 
     with gr.Group():
         model_chooser = gr.Radio(label='Choose Your Model', choices=list(model_info.keys()), value=model_name)
-    
+        welcome_model = gr.Markdown(welcome_message.format(model_name=model_name))
     with gr.Blocks() as demo_blocks:
         gr.Markdown('## Choose Your Interpretation Prompt')
         with gr.Group('Interpretation'):
@@ -221,6 +222,6 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
 
     extra_components = [interpretation_prompt, original_prompt_raw, 
                         original_prompt_btn, *interpretation_bubbles]
-    model_chooser.change(reset_model, [model_chooser, *extra_components], extra_components)
+    model_chooser.change(reset_model, [model_chooser, *extra_components], [welcome_model, *extra_components])
     
     demo.launch()