Merge pull request #535 from mlrun/1.5.x-dev

1.5.x dev
mlrun · Oct 31, 2023 · 4b317c0 · 4b317c0
2 parents 3a8f923 + d5c91ab
commit 4b317c0
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",