fashxp commited on
Commit
50a101c
·
1 Parent(s): 2d4a49c
Files changed (1) hide show
  1. 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]