TalHach61 commited on
Commit
22b3efc
1 Parent(s): 1801f11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def resize_image(image):
52
  def generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditioning_scale, seed):
53
  generator = torch.Generator("cuda").manual_seed(seed)
54
  images = pipe(
55
- prompt, negative_prompt=negative_prompt, image=grayscale_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
56
  generator=generator,
57
  ).images
58
  return images
 
52
  def generate_(prompt, negative_prompt, pose_image, num_steps, controlnet_conditioning_scale, seed):
53
  generator = torch.Generator("cuda").manual_seed(seed)
54
  images = pipe(
55
+ prompt, negative_prompt=negative_prompt, image=pose_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
56
  generator=generator,
57
  ).images
58
  return images