RamAnanth1 commited on
Commit
d88097b
·
1 Parent(s): b75b8f0

fix output type in canny

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -268,7 +268,7 @@ class canny2image:
268
  output = self.pipe(prompt,canny_image,generator=self.generator,num_images_per_prompt=1,num_inference_steps=20)
269
 
270
  updated_image_path = get_new_image_name(image_path, func_name="canny2image")
271
- real_image = Image.fromarray(output.images[0]) # get default the index0 image
272
  real_image.save(updated_image_path)
273
  return updated_image_path
274
 
 
268
  output = self.pipe(prompt,canny_image,generator=self.generator,num_images_per_prompt=1,num_inference_steps=20)
269
 
270
  updated_image_path = get_new_image_name(image_path, func_name="canny2image")
271
+ real_image = output.images[0] # get default the index0 image
272
  real_image.save(updated_image_path)
273
  return updated_image_path
274