diff --git a/api/onnx_web/output.py b/api/onnx_web/output.py index ec293d66..745f0ad8 100644 --- a/api/onnx_web/output.py +++ b/api/onnx_web/output.py @@ -131,7 +131,7 @@ def save_image( exif = PngImagePlugin.PngInfo() if params is not None: - exif.add_text("Parameters", str_params([output], params, size)) + exif.add_text("Parameters", str_params(params, size)) exif.add_text( "JSON Parameters", json_params( @@ -150,7 +150,7 @@ def save_image( { "0th": { ExifIFD.UserComment: UserComment.dump( - str_params([output], params, size), encoding="unicode" + str_params(params, size), encoding="unicode" ), ImageIFD.Make: "onnx-web", ImageIFD.Model: "TODO", diff --git a/onnx-web.code-workspace b/onnx-web.code-workspace index 977dacee..e190a56e 100644 --- a/onnx-web.code-workspace +++ b/onnx-web.code-workspace @@ -39,6 +39,7 @@ "Dreambooth", "dtype", "ESRGAN", + "Exif", "ftfy", "gfpgan", "Heun",