karimbenharrak
commited on
Commit
•
79bb82a
1
Parent(s):
a678ae0
Update handler.py
Browse files- handler.py +4 -1
handler.py
CHANGED
@@ -41,6 +41,8 @@ class EndpointHandler():
|
|
41 |
encoded_mask_image = data.pop("mask_image", None)
|
42 |
|
43 |
prompt = data.pop("prompt", "")
|
|
|
|
|
44 |
|
45 |
# process image
|
46 |
if encoded_image is not None and encoded_mask_image is not None:
|
@@ -53,7 +55,7 @@ class EndpointHandler():
|
|
53 |
self.pipe.enable_xformers_memory_efficient_attention()
|
54 |
|
55 |
# run inference pipeline
|
56 |
-
out = self.pipe(prompt=prompt, image=image, mask_image=mask_image)
|
57 |
|
58 |
image = out.images[0].resize((1024, 1024))
|
59 |
|
@@ -61,6 +63,7 @@ class EndpointHandler():
|
|
61 |
|
62 |
image = self.pipe2(
|
63 |
prompt=prompt,
|
|
|
64 |
image=image,
|
65 |
mask_image=mask_image,
|
66 |
guidance_scale=8.0,
|
|
|
41 |
encoded_mask_image = data.pop("mask_image", None)
|
42 |
|
43 |
prompt = data.pop("prompt", "")
|
44 |
+
|
45 |
+
negative_prompt = data.pop("negative_prompt", "")
|
46 |
|
47 |
# process image
|
48 |
if encoded_image is not None and encoded_mask_image is not None:
|
|
|
55 |
self.pipe.enable_xformers_memory_efficient_attention()
|
56 |
|
57 |
# run inference pipeline
|
58 |
+
out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image)
|
59 |
|
60 |
image = out.images[0].resize((1024, 1024))
|
61 |
|
|
|
63 |
|
64 |
image = self.pipe2(
|
65 |
prompt=prompt,
|
66 |
+
negative_prompt=negative_prompt,
|
67 |
image=image,
|
68 |
mask_image=mask_image,
|
69 |
guidance_scale=8.0,
|