Spaces:
Running
Running
ciover2024
commited on
Commit
•
2f6f262
1
Parent(s):
d1ff975
Update app.py
Browse files
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
|
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
|