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,
|