Spaces:
Runtime error
Runtime error
Commit
•
933cc37
1
Parent(s):
4e30dae
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def predict(radio, dict, word_mask, prompt=""):
|
|
59 |
mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
|
60 |
os.remove(filename)
|
61 |
#with autocast("cuda"):
|
62 |
-
output = pipe(prompt = prompt,
|
63 |
return output.images[0]
|
64 |
|
65 |
# examples = [[dict(image="init_image.png", mask="mask_image.png"), "A panda sitting on a bench"]]
|
|
|
59 |
mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
|
60 |
os.remove(filename)
|
61 |
#with autocast("cuda"):
|
62 |
+
output = pipe(prompt = prompt, image=init_image, mask_image=mask, strength=0.8)
|
63 |
return output.images[0]
|
64 |
|
65 |
# examples = [[dict(image="init_image.png", mask="mask_image.png"), "A panda sitting on a bench"]]
|