update ui

app.py

@@ -4394,8 +4394,7 @@ with demo:
 
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
-                add_download_button(input_gallery)
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section(allow_download=True)
                 num_images_slider.value = 30
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information", autofocus=False, autoscroll=False)