Update app.py

app.py

@@ -189,8 +189,33 @@ 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, post_processed_result])
-    post_process_button.click(fn=postprocess, inputs=[post_processed_result, enabled, downscale, need_rescale, enable_color_limit, number_of_colors, quantization_method, dither_method, use_k_means, is_grayscale, number_of_shades, quantization_method_grayscale, dither_method_grayscale, use_k_means_grayscale, is_black_and_white, is_inversed_black_and_white, black_and_white_threshold, use_color_palette, palette_image, palette_colors, dither_method_palette], outputs=[post_processed_result])
-
+    post_process_button.click(
+    fn=postprocess, 
+    inputs=[
+        post_processed_result, 
+        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
+    ], 
+    outputs=[post_processed_result]
+)
     
     app.queue(max_size=20, concurrency_count=5)
     app.launch()