add category

app_base.py

@@ -110,6 +110,7 @@ def create_demo() -> gr.Blocks:
                 edit_prompt = gr.Textbox(lines=1, label="Edit Prompt", value=DEFAULT_EDIT_PROMPT)
                 brightness = gr.Slider(minimum=0, maximum=2, value=1.0, step=0.1, label="Brightness")
                 with gr.Accordion("Advanced Options", open=False):
+                    category = gr.Textbox(label="Category", value=DEFAULT_CATEGORY, visible=False)
                     mask_expansion = gr.Number(label="Mask Expansion", value=50, visible=True)
                     mask_dilation = gr.Slider(minimum=0, maximum=10, value=2, step=1, label="Mask Dilation")
                     save_quality = gr.Slider(minimum=1, maximum=100, value=95, step=1, label="Save Quality")
@@ -135,15 +136,15 @@ def create_demo() -> gr.Blocks:
 
         g_btn.click(
             fn=segment_image,
-            inputs=[input_image, DEFAULT_CATEGORY, generate_size, mask_expansion, mask_dilation],
+            inputs=[input_image, category, generate_size, mask_expansion, mask_dilation],
             outputs=[origin_area_image, cropper],
         ).success(
             fn=image_to_image,
-            inputs=[origin_area_image, input_image_prompt, edit_prompt,seed,w1, num_steps, start_step, guidance_scale],
+            inputs=[origin_area_image, input_image_prompt, edit_prompt,seed,w1, num_steps, start_step, guidance_scale, brightness],
             outputs=[enhanced_image, generated_cost],
         ).success(
             fn=restore_result_and_save,
-            inputs=[cropper, DEFAULT_CATEGORY, enhanced_image, save_quality],
+            inputs=[cropper, category, enhanced_image, save_quality],
             outputs=[restored_image, download_path],
         )