Update app.py

app.py

@@ -155,7 +155,7 @@ with gr.Blocks(fill_width=True, head=js, css=css) as demo:
                 # gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3)
                 butState = gr.State(value="Off")
                 gen_button = gr.Button(value="Off", variant="secondary", elem_classes=["off"], scale=3)
-                gen_button.click(fn=toggle_button, inputs=butState, outputs=[butState, button])
+                gen_button.click(fn=toggle_button, inputs=butState, outputs=[butState, gen_button])
                 
                 random_button = gr.Button(f'Randomize Models', variant='secondary', scale=1)