Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
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"]]
|