Spaces:
Runtime error
Runtime error
Commit
·
d88097b
1
Parent(s):
b75b8f0
fix output type in canny
Browse files
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 =
|
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 |
|