Update app.py

app.py

@@ -43,12 +43,6 @@ for act in os.listdir(model_root):
 
 activations = sorted(activations)
 
-def update_seeds(activation):
-    return {"choices": seeds_dict[activation], "value": seeds_dict[activation][0]}
-
-def update_neurons(activation, seed):
-    neurons = neurons_dict[(activation, seed)]
-    return {"choices": neurons, "value": neurons[0]}
 
 # Step 3: Load linear models
 # linear_rgb_path = os.path.join(unzip_dir, "linear_models/linear_rgb.joblib")
@@ -81,11 +75,11 @@ def predict(r, g, b, activation, seed, neurons):
 # Dynamic components for UI
 
 def update_seeds(activation):
-    return gr.Dropdown.update(choices=seeds_dict[activation], value=seeds_dict[activation][0])
+    return {"choices": seeds_dict[activation], "value": seeds_dict[activation][0]}
 
 def update_neurons(activation, seed):
     neurons = neurons_dict[(activation, seed)]
-    return gr.Dropdown.update(choices=neurons, value=neurons[0])
+    return {"choices": neurons, "value": neurons[0]}
 
 # Gradio Interface
 with gr.Blocks() as demo: