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