Merge pull request #536 from aviaIguazio/mask-detection

Mask detection - update pipeline requirements
mlrun · Nov 7, 2023 · 460adde · 460adde
2 parents d227f29 + 8ee891d
commit 460adde
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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",