add a few extra metadata recognizers

Stability-AI · Nov 16, 2023 · 948ccbc · 948ccbc
1 parent 5724d0f
commit 948ccbc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/wwwroot/js/genpage/main.js b/src/wwwroot/js/genpage/main.js
@@ -1118,7 +1118,7 @@ function remapMetadataKeys(metadata, keymap) {
     return metadata;
 }
 
-const imageMetadataKeys = ['prompt', 'Prompt', 'parameters', 'Parameters', 'userComment', 'UserComment'];
+const imageMetadataKeys = ['prompt', 'Prompt', 'parameters', 'Parameters', 'userComment', 'UserComment', 'model', 'Model'];
 
 function imageInputHandler() {
     let imageArea = getRequiredElementById('current_image');
@@ -1159,6 +1159,12 @@ function imageInputHandler() {
                             else if (parsed.userComment) {
                                 metadata = parsed.userComment;
                             }
+                            else if (parsed.model) {
+                                metadata = parsed.model;
+                            }
+                            else if (parsed.Model) {
+                                metadata = parsed.Model;
+                            }
                         }
                         if (metadata instanceof Uint8Array) {
                             let prefix = metadata.slice(0, 8);