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

1.5.x dev
mlrun · Oct 4, 2023 · 3a8f923 · 3a8f923
2 parents 6925786 + ca320ba
commit 3a8f923
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
@@ -214,7 +214,12 @@
     "\n",
     "\n",
     "# Import the `onnx_utils` function to optimize the model:\n",
-    "project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\")\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",
+    "\n",
+    "onnx_func.deploy()\n",
+    "\n",
+    "onnx_func.save()"
    ]
   },
   {