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])
         |