Update app.py

app.py

@@ -48,13 +48,13 @@ with gr.Blocks() as demo:
         )
         task_selector = gr.Dropdown(
             choices=[
-                "Layer wise non-linearity (with first layer)", 
+                "Layer wise non-linearity", 
                 "Next-token prediction from intermediate representations", 
                 "Contextualization measurement",
                 "Layerwise predictions (logit lens)",
                 "Tokenwise loss without i-th layer"
             ], 
-            value="Layer wise non-linearity (with first layer)",
+            value="Layer wise non-linearity",
             label="Select Mode"
         )
         normalization_selector = gr.Dropdown(
@@ -74,7 +74,7 @@ with gr.Blocks() as demo:
             return img, combined_log
         
         def set_default(task_name: str) -> str:
-            if task_name == "Layer wise non-linearity (with first layer)":
+            if task_name == "Layer wise non-linearity":
                 return "token-wise"
             if task_name == "Next-token prediction from intermediate representations":
                 return "token-wise"