apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): helper to get borders from request

This commit is contained in:
Sean Sube 2023-01-16 07:45:50 -06:00
parent 53fcc0b91f
commit 604cdf39b0
1 changed files with 20 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
api/onnx_web/serve.py
View File

@ -173,6 +173,19 @@ def pipeline_from_request() -> Tuple[BaseParams, Size]:
return (params, size)
def border_from_request() -> Border:
left = get_and_clamp_int(request.args, 'left', 0,
config_params.get('width').get('max'), 0)
right = get_and_clamp_int(request.args, 'right',
0, config_params.get('width').get('max'), 0)
top = get_and_clamp_int(request.args, 'top', 0,
config_params.get('height').get('max'), 0)
bottom = get_and_clamp_int(
request.args, 'bottom', 0, config_params.get('height').get('max'), 0)
return Border(left, right, top, bottom)
def check_paths():
if not path.exists(model_path):
raise RuntimeError('model path must exist')
@ -264,14 +277,14 @@ def img2img():
source_file = request.files.get('source')
source_image = Image.open(BytesIO(source_file.read())).convert('RGB')
params, size = pipeline_from_request()
strength = get_and_clamp_float(
request.args,
'strength',
config_params.get('strength').get('default'),
config_params.get('strength').get('max'))
params, size = pipeline_from_request()
output = make_output_name(
'img2img',
params,
@ -320,16 +333,7 @@ def inpaint():
mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB')
params, size = pipeline_from_request()
left = get_and_clamp_int(request.args, 'left', 0,
config_params.get('width').get('max'), 0)
right = get_and_clamp_int(request.args, 'right',
0, config_params.get('width').get('max'), 0)
top = get_and_clamp_int(request.args, 'top', 0,
config_params.get('height').get('max'), 0)
bottom = get_and_clamp_int(
request.args, 'bottom', 0, config_params.get('height').get('max'), 0)
expand = Border(left, right, top, bottom)
expand = border_from_request()
mask_filter = get_from_map(request.args, 'filter', mask_filters, 'none')
noise_source = get_from_map(
@ -341,10 +345,10 @@ def inpaint():
params,
size,
extras=(
left,
right,
top,
bottom,
expand.left,
expand.right,
expand.top,
expand.bottom,
mask_filter.__name__,
noise_source.__name__,
)