MaxMilan1
commited on
Commit
·
48a21a5
1
Parent(s):
1949e8b
change so that image is returned as a PIL image
Browse files- util/text_img.py +1 -0
util/text_img.py
CHANGED
|
@@ -44,6 +44,7 @@ def generate_image(prompt, negative_prompt, control_image, scale=0.5):
|
|
| 44 |
controlnet_conditioning_scale=scale,
|
| 45 |
).images[0]
|
| 46 |
image2 = rembg.remove(image)
|
|
|
|
| 47 |
|
| 48 |
return image2
|
| 49 |
|
|
|
|
| 44 |
controlnet_conditioning_scale=scale,
|
| 45 |
).images[0]
|
| 46 |
image2 = rembg.remove(image)
|
| 47 |
+
image2 = Image.fromarray(image2)
|
| 48 |
|
| 49 |
return image2
|
| 50 |
|