apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): send correct image URLs

This commit is contained in:
Sean Sube 2024-01-28 20:04:47 -06:00
parent 8c8be8fc08
commit 8b26c12439
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
api/onnx_web/output.py
View File

@ -70,11 +70,11 @@ def save_result(
save_thumbnails: bool = False,
) -> List[str]:
images = result.as_images()
output_names = make_output_names(server, base_name, len(images))
logger.debug("saving %s images: %s", len(images), output_names)
result.outputs = make_output_names(server, base_name, len(images))
logger.debug("saving %s images: %s", len(images), result.outputs)
outputs = []
for image, metadata, filename in zip(images, result.metadata, output_names):
for image, metadata, filename in zip(images, result.metadata, result.outputs):
outputs.append(
save_image(
server,
@ -84,19 +84,17 @@ def save_result(
)
)
result.outputs = outputs
if save_thumbnails:
thumbnail_names = make_output_names(
result.thumbnails = make_output_names(
server,
base_name,
len(images),
suffix="thumbnail",
)
logger.debug("saving %s thumbnails: %s", len(images), thumbnail_names)
logger.debug("saving %s thumbnails: %s", len(images), result.thumbnails)
thumbnails = []
for image, filename in zip(images, thumbnail_names):
for image, filename in zip(images, result.thumbnails):
thumbnail = image.copy()
thumbnail.thumbnail((server.thumbnail_size, server.thumbnail_size))
@ -108,8 +106,6 @@ def save_result(
)
)
result.thumbnails = thumbnails
return outputs