apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add size

This commit is contained in:
Sean Sube 2023-09-10 23:20:49 -05:00
parent b0df6247b2
commit 0047e1a12c
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/chain/persist_disk.py
View File

@ -4,7 +4,7 @@ from typing import List, Optional
from PIL import Image
from ..output import save_image
from ..params import ImageParams, StageParams
from ..params import ImageParams, Size, StageParams
from ..server import ServerContext
from ..worker import WorkerContext
from .stage import BaseStage
@ -22,11 +22,12 @@ class PersistDiskStage(BaseStage):
sources: List[Image.Image],
*,
output: List[str],
size: Optional[Size] = None,
stage_source: Optional[Image.Image] = None,
**kwargs,
) -> List[Image.Image]:
for source, name in zip(sources, output):
dest = save_image(server, name, source, params=params)
dest = save_image(server, name, source, params=params, size=size)
logger.info("saved image to %s", dest)
return sources