apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

extract device

This commit is contained in:
Sean Sube 2023-02-04 15:00:22 -06:00
parent 59f5fd41b0
commit 1e38659c80
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/serve.py
View File

@ -441,7 +441,7 @@ def img2img():
source_file = request.files.get('source') source_file = request.files.get('source')
source_image = Image.open(BytesIO(source_file.read())).convert('RGB') source_image = Image.open(BytesIO(source_file.read())).convert('RGB')
params, size = pipeline_from_request() device, params, size = pipeline_from_request()
upscale = upscale_from_request(params.provider) upscale = upscale_from_request(params.provider)
strength = get_and_clamp_float( strength = get_and_clamp_float(
@ -468,7 +468,7 @@ def img2img():
@app.route('/api/txt2img', methods=['POST']) @app.route('/api/txt2img', methods=['POST'])
def txt2img(): def txt2img():
params, size = pipeline_from_request() device, params, size = pipeline_from_request()
upscale = upscale_from_request(params.provider) upscale = upscale_from_request(params.provider)
output = make_output_name( output = make_output_name(
@ -498,7 +498,7 @@ def inpaint():
mask_file = request.files.get('mask') mask_file = request.files.get('mask')
mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB') mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB')
params, size = pipeline_from_request() device, params, size = pipeline_from_request()
expand = border_from_request() expand = border_from_request()
upscale = upscale_from_request(params.provider) upscale = upscale_from_request(params.provider)
@ -560,7 +560,7 @@ def upscale():
source_file = request.files.get('source') source_file = request.files.get('source')
source_image = Image.open(BytesIO(source_file.read())).convert('RGB') source_image = Image.open(BytesIO(source_file.read())).convert('RGB')
params, size = pipeline_from_request() device, params, size = pipeline_from_request()
upscale = upscale_from_request(params.provider) upscale = upscale_from_request(params.provider)
output = make_output_name( output = make_output_name(
@ -593,7 +593,7 @@ def chain():
validate(data, schema) validate(data, schema)
# get defaults from the regular parameters # get defaults from the regular parameters
params, size = pipeline_from_request() device, params, size = pipeline_from_request()
output = make_output_name( output = make_output_name(
context, context,
'chain', 'chain',