ciover2024 commited on
Commit
5aa8d79
·
verified ·
1 Parent(s): 72c7585

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -111,7 +111,7 @@ def save_mask(mask_local_path, save_path="generated_mask.png"):
111
 
112
  # Function to perform inpainting
113
  def inpaint_image(input_image, mask_image):
114
- prompt = "undress, naked"
115
  negative_prompt = "bad anatomy, deformed, ugly, disfigured"
116
  """
117
  IMAGE_SIZE = (1024,1024)
@@ -120,8 +120,8 @@ def inpaint_image(input_image, mask_image):
120
  blurred_mask_image = inpaint_pipeline.mask_processor.blur(initial_mask_image,blur_factor=50)
121
  result = inpaint_pipeline(prompt=prompt, negative_prompt=negative_prompt, height=IMAGE_SIZE[0], width=IMAGE_SIZE[0], image=initial_input_image, mask_image=blurred_mask_image, padding_mask_crop=32)
122
  """
123
- blurred_mask_image = inpaint_pipeline.mask_processor.blur(mask_image,blur_factor=50)
124
- result = inpaint_pipeline(prompt=prompt, negative_prompt=negative_prompt, image=input_image, mask_image=blurred_mask_image, padding_mask_crop=32)
125
  inpainted_image = result.images[0]
126
  #inpainted_image = resize_to_match(input_image, inpainted_image)
127
  return inpainted_image
 
111
 
112
  # Function to perform inpainting
113
  def inpaint_image(input_image, mask_image):
114
+ prompt = "undress, naked, real skin, ultra detailed nipple, ultra detailed shaved pussy"
115
  negative_prompt = "bad anatomy, deformed, ugly, disfigured"
116
  """
117
  IMAGE_SIZE = (1024,1024)
 
120
  blurred_mask_image = inpaint_pipeline.mask_processor.blur(initial_mask_image,blur_factor=50)
121
  result = inpaint_pipeline(prompt=prompt, negative_prompt=negative_prompt, height=IMAGE_SIZE[0], width=IMAGE_SIZE[0], image=initial_input_image, mask_image=blurred_mask_image, padding_mask_crop=32)
122
  """
123
+ #blurred_mask_image = inpaint_pipeline.mask_processor.blur(mask_image,blur_factor=50)
124
+ result = inpaint_pipeline(prompt=prompt, negative_prompt=negative_prompt, image=input_image, mask_image=mask_image, padding_mask_crop=32)
125
  inpainted_image = result.images[0]
126
  #inpainted_image = resize_to_match(input_image, inpainted_image)
127
  return inpainted_image