apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

select correct platform

This commit is contained in:
Sean Sube 2023-02-11 16:26:00 -06:00
parent 66c2ce3074
commit 6d243e8f4f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/serve.py
View File

@ -163,7 +163,7 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]:
if device_name is not None and device_name != "any":
for platform in available_platforms:
if platform.device == device_name:
device = available_platforms[0]
device = platform
# pipeline stuff
lpw = get_not_empty(request.args, "lpw", "false") == "true"