diff --git a/mask-detection/3-automatic-pipeline.ipynb b/mask-detection/3-automatic-pipeline.ipynb index eb6d5d56..6a957696 100644 --- a/mask-detection/3-automatic-pipeline.ipynb +++ b/mask-detection/3-automatic-pipeline.ipynb @@ -215,7 +215,12 @@ "\n", "# Import the `onnx_utils` function to optimize the model:\n", "# Import the `onnx_utils` function to optimize the model:\n", - "onnx_func = project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\",requirements=['tensorflow==2.9.0'] if framework=='tf-keras' else ['torch'])\n", + "onnx_func = project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\",requirements=['tensorflow==2.9.0',\n", + " 'onnx~=1.10.1',\n", + " 'onnxruntime~=1.8.1',\n", + " 'onnxoptimizer~=0.2.0',\n", + " 'onnxmltools~=1.9.0',\n", + " 'tf2onnx~=1.9.0'] if framework=='tf-keras' else ['torch'])\n", "\n", "onnx_func.deploy()\n", "\n",