fix(api): use first available platform if requested one does not exist

ssube · Feb 5, 2023 · 15c651b · 15c651b
1 parent 401df84
commit 15c651b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -168,7 +168,8 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]:
             device = available_platforms[0]
 
     if device is None:
-        raise Exception("unknown device")
+        logger.warn('unknown platform: %s', device_name)
+        device = available_platforms[0]
 
     # pipeline stuff
     model = get_not_empty(request.args, "model", get_config_value("model"))