Update app.py

app.py

@@ -191,31 +191,31 @@ with gr.Blocks() as app:
     # The rest of your code for setting up the app
     gallery.select(update_selection, outputs=[prompt, selected_info, selected_state])
     prompt.submit(fn=run_lora, inputs=[prompt, selected_state], outputs=[result])
-    button.click(fn=run_lora, inputs=[prompt, selected_state], outputs=[result])
-    post_process_button.click(
-    fn=postprocess, 
+    button.click(
+    fn=run_and_postprocess, 
     inputs=[
-        result, 
+        prompt, 
+        selected_state, 
         enabled, 
         downscale, 
         need_rescale, 
         enable_color_limit, 
-        palette_size_color,  # Updated this
-        quantization_methods_color,  # Updated this
-        dither_methods_color,  # Updated this
-        k_means_color,  # Updated this
-        enable_grayscale,  # Updated this
-        palette_size_gray,  # Updated this
-        quantization_methods_gray,  # Updated this
-        dither_methods_gray,  # Updated this
-        k_means_gray,  # Updated this
-        enable_black_and_white,  # Updated this
-        inverse_black_and_white,  # Updated this
-        threshold_black_and_white,  # Updated this
-        enable_custom_palette,  # Updated this
-        palette_image,  # Updated this
-        palette_size_custom,  # Updated this
-        dither_methods_custom  # Updated this
+        palette_size_color, 
+        quantization_methods_color, 
+        dither_methods_color, 
+        k_means_color, 
+        enable_grayscale, 
+        palette_size_gray, 
+        quantization_methods_gray, 
+        dither_methods_gray, 
+        k_means_gray, 
+        enable_black_and_white, 
+        inverse_black_and_white, 
+        threshold_black_and_white, 
+        enable_custom_palette, 
+        palette_image, 
+        palette_size_custom, 
+        dither_methods_custom
     ], 
     outputs=[result]
 )