Restore activation threshold parameter for random neuron discovery

app.py

@@ -147,11 +147,11 @@ with gr.Blocks() as demo:
             neuron_index = gr.Number(
                 label="Neuron Index", value=default_neuron_index, precision=0
             )
+            max_act = gr.Number(label="Max Activation Threshold", value=3.0, precision=2)
             random_btn = gr.Button("Find Random Active Neuron")
-            max_act = gr.Number(label="Max Activation", value=default_layer, precision=0)
             max_val = gr.Number(label="Max Value", value=default_max_val)
             min_val = gr.Number(label="Min Value", value=default_min_val)
-            inputs = [text, layer, neuron_index, max_val, min_val, max_act]
+            inputs = [text, layer, neuron_index, max_val, min_val]
         with gr.Column():
             out = gr.HTML(
                 label="Neuron Acts",
@@ -164,13 +164,13 @@ with gr.Blocks() as demo:
                 ),
             )
     
-    def random_neuron_callback(text):
-        layer_num, neuron_num = get_random_active_neuron(text)
+    def random_neuron_callback(text, threshold):
+        layer_num, neuron_num = get_random_active_neuron(text, threshold)
         return layer_num, neuron_num
     
     random_btn.click(
         random_neuron_callback,
-        inputs=[text],
+        inputs=[text, max_act],
         outputs=[layer, neuron_index]
     )