xichenhku commited on
Commit
8bb5e11
1 Parent(s): aaa2028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -184,7 +184,7 @@ mimicbrush_model = MimicBrush_RefNet(pipe, image_encoder_path, mimicbrush_ckpt,
184
  mask_processor = VaeImageProcessor(vae_scale_factor=1, do_normalize=False, do_binarize=True, do_convert_grayscale=True)
185
 
186
  @spaces.GPU
187
- def infer_single(ref_image, target_image, target_mask, seed = -1, num_inference_steps=50, guidance_scale = 5, enable_shape_control = False, progress=gr.Progress(track_tqdm=True)):
188
  #return ref_image
189
  """
190
  mask: 0/1 1-channel np.array
@@ -314,7 +314,7 @@ with gr.Blocks() as demo:
314
  with gr.Row():
315
  base = gr.ImageEditor( label="Source",
316
  type="pil",
317
- brush=gr.Brush(colors=["#80FFFF"],default_size = 30,color_mode = "fixed"),
318
  layers = False,
319
  interactive=True
320
  )
 
184
  mask_processor = VaeImageProcessor(vae_scale_factor=1, do_normalize=False, do_binarize=True, do_convert_grayscale=True)
185
 
186
  @spaces.GPU
187
+ def infer_single(ref_image, target_image, target_mask, seed = -1, num_inference_steps=50, guidance_scale = 5, enable_shape_control = False):
188
  #return ref_image
189
  """
190
  mask: 0/1 1-channel np.array
 
314
  with gr.Row():
315
  base = gr.ImageEditor( label="Source",
316
  type="pil",
317
+ brush=gr.Brush(colors=["#FFFFFF"],default_size = 30,color_mode = "fixed"),
318
  layers = False,
319
  interactive=True
320
  )