Spaces:
Runtime error
Runtime error
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])
|