jhj0517 commited on
Commit
ee4969b
Β·
1 Parent(s): 6971c54

Add `invert_mask` component

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -132,6 +132,7 @@ class App:
132
  nb_pixel_size = gr.Number(label="Pixel Size", interactive=True, minimum=1,
133
  visible=self.default_filter == PIXELIZE_FILTER,
134
  value=self.default_pixel_size)
 
135
  btn_generate_preview = gr.Button("GENERATE PREVIEW")
136
 
137
  with gr.Row():
@@ -157,7 +158,7 @@ class App:
157
  nb_pixel_size])
158
 
159
  preview_params = [vid_frame_prompter, dd_filter_mode, sld_frame_selector, nb_pixel_size,
160
- cp_color_picker]
161
  btn_generate_preview.click(fn=self.sam_inf.add_filter_to_preview,
162
  inputs=preview_params,
163
  outputs=[img_preview])
@@ -180,6 +181,7 @@ class App:
180
  choices=self.image_modes)
181
  dd_models = gr.Dropdown(label="Model", value=DEFAULT_MODEL_TYPE,
182
  choices=self.sam_inf.available_models)
 
183
 
184
  with gr.Accordion("Mask Parameters", open=False, visible=self.default_mode == AUTOMATIC_MODE) as acc_mask_hparams:
185
  mask_hparams_component = self.mask_generation_parameters(_mask_hparams)
@@ -194,10 +196,9 @@ class App:
194
  output_file = gr.File(label="Generated psd file", scale=9)
195
  btn_open_folder = gr.Button("πŸ“\nOpen PSD folder", scale=1)
196
 
197
- sources = [img_input, img_input_prompter, dd_input_modes]
198
- model_params = [dd_models]
199
  mask_hparams = mask_hparams_component + [cb_multimask_output]
200
- input_params = sources + model_params + mask_hparams
201
 
202
  btn_generate.click(fn=self.sam_inf.divide_layer,
203
  inputs=input_params, outputs=[gallery_output, output_file])
 
132
  nb_pixel_size = gr.Number(label="Pixel Size", interactive=True, minimum=1,
133
  visible=self.default_filter == PIXELIZE_FILTER,
134
  value=self.default_pixel_size)
135
+ cb_invert_mask = gr.Checkbox(label="invert mask", value=_mask_hparams["invert_mask"])
136
  btn_generate_preview = gr.Button("GENERATE PREVIEW")
137
 
138
  with gr.Row():
 
158
  nb_pixel_size])
159
 
160
  preview_params = [vid_frame_prompter, dd_filter_mode, sld_frame_selector, nb_pixel_size,
161
+ cp_color_picker, cb_invert_mask]
162
  btn_generate_preview.click(fn=self.sam_inf.add_filter_to_preview,
163
  inputs=preview_params,
164
  outputs=[img_preview])
 
181
  choices=self.image_modes)
182
  dd_models = gr.Dropdown(label="Model", value=DEFAULT_MODEL_TYPE,
183
  choices=self.sam_inf.available_models)
184
+ cb_invert_mask = gr.Checkbox(label="invert mask", value=_mask_hparams["invert_mask"])
185
 
186
  with gr.Accordion("Mask Parameters", open=False, visible=self.default_mode == AUTOMATIC_MODE) as acc_mask_hparams:
187
  mask_hparams_component = self.mask_generation_parameters(_mask_hparams)
 
196
  output_file = gr.File(label="Generated psd file", scale=9)
197
  btn_open_folder = gr.Button("πŸ“\nOpen PSD folder", scale=1)
198
 
199
+ input_params = [img_input, img_input_prompter, dd_input_modes, dd_models, cb_invert_mask]
 
200
  mask_hparams = mask_hparams_component + [cb_multimask_output]
201
+ input_params += mask_hparams
202
 
203
  btn_generate.click(fn=self.sam_inf.divide_layer,
204
  inputs=input_params, outputs=[gallery_output, output_file])