alfabill commited on
Commit
33fd751
·
1 Parent(s): 7528b73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -65,7 +65,8 @@ def predict(radio, dict, word_mask, prompt=""):
65
  mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
66
  os.remove(filename)
67
  #with autocast("cuda"):
68
- output = pipe(prompt = prompt, image=init_image, mask_image=mask, num_inference_steps=6)
 
69
  return output.images[0]
70
 
71
  # examples = [[dict(image="init_image.png", mask="mask_image.png"), "A panda sitting on a bench"]]
 
65
  mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
66
  os.remove(filename)
67
  #with autocast("cuda"):
68
+ negative="tainted face, tainted skin, tattoo, cloth, clothed, sfw, artistic, cartoon, unrealistic, sfw, half body, paint, draw, 3d, art,illustration, cropped, low quality, blur, noise, chalk, anime, b&w, sfw, malformations, allucinations, errors, art, draw, 3d, comic, low quality, lowres, bad anatomy, bad hands, cropped, worst quality, openjourney, low quality, worst quality, bad anatomy, bad proportions"
69
+ output = pipe(prompt = prompt, image=init_image, mask_image=mask, num_inference_steps=6, negative_prompt=negative)
70
  return output.images[0]
71
 
72
  # examples = [[dict(image="init_image.png", mask="mask_image.png"), "A panda sitting on a bench"]]