Update app.py

app.py

@@ -478,8 +478,8 @@ with gr.Blocks(css="style.css") as demo:
     
                 with gr.Column():
                     gr.Markdown("""❷ Generate new images of the sampled/inverted identity & edit with the sliders""")
-                    #gallery = gr.Image(label="Generated Image",height=512, width=512, interactive=False)
-                    image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
+                    gallery = gr.Image(label="Generated Image",height=512, width=512, interactive=False)
+                    #image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
                     submit = gr.Button("Generate")
     
                         
@@ -543,7 +543,7 @@ with gr.Blocks(css="style.css") as demo:
         sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image])
             
         submit.click(
-                fn=edit_inference, inputs=[net, prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4, input_image], outputs=[net, image_slider]
+                fn=edit_inference, inputs=[net, prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4, input_image], outputs=[net, gallery]
             )
         file_input.change(fn=file_upload, inputs=[file_input, net], outputs = [net, input_image])