update
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def predict(dict, prompt=""):
|
|
38 |
with autocast("cuda"):
|
39 |
init_image = dict["image"].convert("RGB").resize((512, 512))
|
40 |
mask = dict["mask"].convert("RGB").resize((512, 512))
|
41 |
-
output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5
|
42 |
return output.images[0]
|
43 |
|
44 |
|
|
|
38 |
with autocast("cuda"):
|
39 |
init_image = dict["image"].convert("RGB").resize((512, 512))
|
40 |
mask = dict["mask"].convert("RGB").resize((512, 512))
|
41 |
+
output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
|
42 |
return output.images[0]
|
43 |
|
44 |
|