Spaces:
Runtime error
Runtime error
fix png?
Browse files
txt2panoimg/text_to_360panorama_image_pipeline.py
CHANGED
|
@@ -207,4 +207,7 @@ class Text2360PanoramaImagePipeline(DiffusionPipeline):
|
|
| 207 |
).images[0]
|
| 208 |
print('finished')
|
| 209 |
|
|
|
|
|
|
|
|
|
|
| 210 |
return output_img
|
|
|
|
| 207 |
).images[0]
|
| 208 |
print('finished')
|
| 209 |
|
| 210 |
+
# Ensure output is in PNG format
|
| 211 |
+
if isinstance(output_img, Image.Image):
|
| 212 |
+
output_img = output_img.convert('RGB')
|
| 213 |
return output_img
|