Update handler.py
Browse files- handler.py +2 -1
handler.py
CHANGED
@@ -108,7 +108,8 @@ class EndpointHandler():
|
|
108 |
out = self.pipe(
|
109 |
prompt=prompt,
|
110 |
negative_prompt=negative_prompt,
|
111 |
-
image=control_image,
|
|
|
112 |
num_inference_steps=num_inference_steps,
|
113 |
guidance_scale=guidance_scale,
|
114 |
num_images_per_prompt=1,
|
|
|
108 |
out = self.pipe(
|
109 |
prompt=prompt,
|
110 |
negative_prompt=negative_prompt,
|
111 |
+
#image=control_image,
|
112 |
+
image=image,
|
113 |
num_inference_steps=num_inference_steps,
|
114 |
guidance_scale=guidance_scale,
|
115 |
num_images_per_prompt=1,
|