Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -197,11 +197,11 @@ def infer(
|
|
| 197 |
path_or_img = item
|
| 198 |
|
| 199 |
if isinstance(path_or_img, str):
|
| 200 |
-
pil_images.append(Image.open(path_or_img).convert("RGB"))
|
| 201 |
elif isinstance(path_or_img, Image.Image):
|
| 202 |
-
pil_images.append(path_or_img.convert("RGB"))
|
| 203 |
else:
|
| 204 |
-
pil_images.append(Image.open(path_or_img.name).convert("RGB"))
|
| 205 |
except Exception as e:
|
| 206 |
print(f"Skipping invalid image item: {e}")
|
| 207 |
continue
|
|
@@ -290,7 +290,7 @@ def infer(
|
|
| 290 |
generator=generator,
|
| 291 |
true_cfg_scale=guidance_scale,
|
| 292 |
).images[0]
|
| 293 |
-
|
| 294 |
return result_image, seed
|
| 295 |
|
| 296 |
except Exception as e:
|
|
|
|
| 197 |
path_or_img = item
|
| 198 |
|
| 199 |
if isinstance(path_or_img, str):
|
| 200 |
+
pil_images.append(Image.open(path_or_img))#.convert("RGB"))
|
| 201 |
elif isinstance(path_or_img, Image.Image):
|
| 202 |
+
pil_images.append(path_or_img)#.convert("RGB"))
|
| 203 |
else:
|
| 204 |
+
pil_images.append(Image.open(path_or_img.name))#.convert("RGB"))
|
| 205 |
except Exception as e:
|
| 206 |
print(f"Skipping invalid image item: {e}")
|
| 207 |
continue
|
|
|
|
| 290 |
generator=generator,
|
| 291 |
true_cfg_scale=guidance_scale,
|
| 292 |
).images[0]
|
| 293 |
+
result_image = result_image.convert("RGB")
|
| 294 |
return result_image, seed
|
| 295 |
|
| 296 |
except Exception as e:
|