Spaces:

cagliostrolab
/

animagine-xl-3.1

Running on Zero

Asahina2K commited on Apr 3

Commit

f240016

•

1 Parent(s): 0cfdbbe

rollback

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -169,7 +169,7 @@ def save_image(image, metadata, output_dir, is_colab):
     filepath = os.path.join(output_dir, filename)
     metadata_str = json.dumps(metadata)
     info = PngImagePlugin.PngInfo()
-    info.add_text("parameters", metadata_str)
     image.save(filepath, "PNG", pnginfo=info)
     return filepath

     filepath = os.path.join(output_dir, filename)
     metadata_str = json.dumps(metadata)
     info = PngImagePlugin.PngInfo()
+    info.add_text("metadata", metadata_str)
     image.save(filepath, "PNG", pnginfo=info)
     return filepath