From 6d243e8f4f3c3d98e455f98f18ae96c25c98a366 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 11 Feb 2023 16:26:00 -0600
Subject: [PATCH] select correct platform

---
 api/onnx_web/serve.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index eb157293..8b0146b7 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -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"