From fe6b26748341d2b11d09659c0494809ea6f6cc66 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Wed, 3 Jan 2024 23:38:44 -0600
Subject: [PATCH] prep models field

---
 api/onnx_web/chain/result.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/api/onnx_web/chain/result.py b/api/onnx_web/chain/result.py
index 93643eba..562188c6 100644
--- a/api/onnx_web/chain/result.py
+++ b/api/onnx_web/chain/result.py
@@ -127,8 +127,15 @@ class ImageMetadata:
         }
 
         # fix up some fields
-        json["params"]["model"] = path.basename(self.params.model)
-        json["params"]["scheduler"] = self.params.scheduler  # TODO: why tho?
+        model = path.basename(self.params.model)
+        json["params"]["model"] = model
+        json["models"].append(
+            {
+                "name": model,
+                "weight": 1.0,
+                "hash": self.get_model_hash(model),
+            }
+        )
 
         # calculate final output size
         output_size = self.size