ciover2024 commited on
Commit
2f6f262
1 Parent(s): d1ff975

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, real skin, detailed nipples, erect nipples, detailed pussy, (detailed nipples), (detailed skin), (detailed pussy)"
115
  negative_prompt = "bad anatomy, deformed, ugly, disfigured, (extra arms), (extra legs), (extra hands), (extra feet), (extra finger)"
116
  """
117
  IMAGE_SIZE = (1024,1024)
@@ -121,7 +121,7 @@ def inpaint_image(input_image, mask_image):
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
 
111
 
112
  # Function to perform inpainting
113
  def inpaint_image(input_image, mask_image):
114
+ prompt = "undress, naked, real skin, detailed nipples, erect nipples, detailed pussy, (detailed nipples), (detailed skin), (detailed pussy), accurate anatomy"
115
  negative_prompt = "bad anatomy, deformed, ugly, disfigured, (extra arms), (extra legs), (extra hands), (extra feet), (extra finger)"
116
  """
117
  IMAGE_SIZE = (1024,1024)
 
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