Update app.py

app.py

@@ -481,10 +481,10 @@ with gr.Blocks(css="style.css") as demo:
 
         invert_button.click(fn=run_inversion,
                     inputs=[input_image, pcs, epochs, weight_decay,lr], 
-                    outputs = [gallery, file_output])
+                    outputs = [input_image, file_output])
         
             
-        sample.click(fn=sample_then_run, outputs=[gallery, file_output])
+        sample.click(fn=sample_then_run, outputs=[input_image, file_output])
 
         
         submit.click(