Spaces:
Running
Running
jhj0517
commited on
Commit
Β·
ee4969b
1
Parent(s):
6971c54
Add `invert_mask` component
Browse files
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 |
-
|
198 |
-
model_params = [dd_models]
|
199 |
mask_hparams = mask_hparams_component + [cb_multimask_output]
|
200 |
-
input_params
|
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])
|