fixing
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -30,5 +30,5 @@ class EndpointHandler():
|
|
30 |
parameters = data.pop("parameters", data)
|
31 |
prompt = parameters['prompt']
|
32 |
|
33 |
-
images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
|
34 |
return images[0]
|
|
|
30 |
parameters = data.pop("parameters", data)
|
31 |
prompt = parameters['prompt']
|
32 |
|
33 |
+
images = self.pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
|
34 |
return images[0]
|