From dc9048c76f60dad0f696ef069f355653e994cfc7 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 2 Aug 2023 17:43:17 -0500 Subject: [PATCH] fix(api): use correct case for PNG metadata --- api/onnx_web/output.py | 8 ++++---- gui/src/components/Profiles.tsx | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/api/onnx_web/output.py b/api/onnx_web/output.py index 6b051854..79fc863c 100644 --- a/api/onnx_web/output.py +++ b/api/onnx_web/output.py @@ -203,9 +203,9 @@ def save_image( exif = PngImagePlugin.PngInfo() if params is not None: - exif.add_text("Make", "onnx-web") + exif.add_text("make", "onnx-web") exif.add_text( - "Maker Note", + "maker note", dumps( json_params( [output], @@ -217,9 +217,9 @@ def save_image( ) ), ) - exif.add_text("Model", server.server_version) + exif.add_text("model", server.server_version) exif.add_text( - "Parameters", + "parameters", str_params(server, params, size, inversions=inversions, loras=loras), ) diff --git a/gui/src/components/Profiles.tsx b/gui/src/components/Profiles.tsx index 84166a92..5ed9de77 100644 --- a/gui/src/components/Profiles.tsx +++ b/gui/src/components/Profiles.tsx @@ -217,7 +217,7 @@ export function downloadParamsAsFile(data: DeepPartial): void { export async function parseImageParams(file: File): Promise> { const tags = await ExifReader.load(file); - // handle lowercase variation from my earlier mistakes + // some parsers expect uppercase, some use lowercase, read both const makerNote = decodeTag(defaultTo(tags.MakerNote, tags['maker note'])); // eslint-disable-next-line dot-notation, @typescript-eslint/strict-boolean-expressions const userComment = decodeTag(defaultTo(defaultTo(tags.UserComment, tags['Parameters']), tags['parameters']));