Update app.py

app.py

@@ -156,8 +156,6 @@ with gr.Blocks() as app:
                 button = gr.Button("Run")
             
             result = gr.Image(interactive=False, label="Generated Image")
-            post_processed_result = gr.Image(interactive=False, label="Post-Processed Image")
-
     # Accordion moved here, inside the same gr.Blocks context
     with gr.Accordion(label="Pixel art", open=True):
         with gr.Row():
@@ -189,8 +187,6 @@ with gr.Blocks() as app:
                 palette_image = gr.Image(label="Color palette image", type="pil")
                 palette_size_custom = gr.Slider(label="Palette Size", minimum=1, maximum=256, step=1, value=16)
                 dither_methods_custom = gr.Radio(choices=["None", "Floyd-Steinberg"], label="Colors Dither Method", value="None")
-                
-                post_process_button = gr.Button("Apply Post-Processing")
 
     # The rest of your code for setting up the app
     gallery.select(update_selection, outputs=[prompt, selected_info, selected_state])
@@ -199,7 +195,7 @@ with gr.Blocks() as app:
     post_process_button.click(
     fn=postprocess, 
     inputs=[
-        post_processed_result, 
+        result, 
         enabled, 
         downscale, 
         need_rescale, 
@@ -221,7 +217,7 @@ with gr.Blocks() as app:
         palette_size_custom,  # Updated this
         dither_methods_custom  # Updated this
     ], 
-    outputs=[post_processed_result]
+    outputs=[result]
 )
     
     app.queue(max_size=20, concurrency_count=5)