dikarel commited on
Commit
1a9bd9f
1 Parent(s): 050d941

attempt to boost quality

Browse files
Files changed (1) hide show
  1. lib/redraw_image.py +2 -1
lib/redraw_image.py CHANGED
@@ -15,11 +15,12 @@ def redraw_image(prompt: str, image: PILImage, mask: PILImage) -> PILImage:
15
 
16
  output = inpaint_model(
17
  prompt=prompt,
 
18
  image=image,
19
  mask_image=mask,
20
  width=image.width,
21
  height=image.height,
22
- num_inference_steps=30,
23
  ).images[0]
24
 
25
  return unpad_image(output, original_image_size)
 
15
 
16
  output = inpaint_model(
17
  prompt=prompt,
18
+ negative_prompt="low quality, low resolution, dark, scary, nsfw",
19
  image=image,
20
  mask_image=mask,
21
  width=image.width,
22
  height=image.height,
23
+ num_inference_steps=60,
24
  ).images[0]
25
 
26
  return unpad_image(output, original_image_size)