diff --git a/mask-detection/3-automatic-pipeline.ipynb b/mask-detection/3-automatic-pipeline.ipynb
index 69130bdd..eb6d5d56 100644
--- 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()"
    ]
   },
   {