Spaces:
Running
on
Zero
Running
on
Zero
Himanshu-AT
commited on
Commit
·
fceb263
1
Parent(s):
61dc46d
update download
Browse files
app.py
CHANGED
|
@@ -111,6 +111,8 @@ def infer(edit_images, prompt, lora_model, seed=42, randomize_seed=False, guidan
|
|
| 111 |
# return image, seed
|
| 112 |
|
| 113 |
def download_image(image):
|
|
|
|
|
|
|
| 114 |
image.save("output.png", "PNG")
|
| 115 |
return "output.png"
|
| 116 |
|
|
|
|
| 111 |
# return image, seed
|
| 112 |
|
| 113 |
def download_image(image):
|
| 114 |
+
if isinstance(image, np.ndarray):
|
| 115 |
+
image = Image.fromarray(image)
|
| 116 |
image.save("output.png", "PNG")
|
| 117 |
return "output.png"
|
| 118 |
|