Skip to content

Commit

Permalink
Merge pull request #529 from aviaIguazio/mask-detection
Browse files Browse the repository at this point in the history
Mask detection
  • Loading branch information
aviaIguazio authored Oct 4, 2023
2 parents 6610d64 + 0918740 commit ca320ba
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion mask-detection/3-automatic-pipeline.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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()"
]
},
{
Expand Down

0 comments on commit ca320ba

Please sign in to comment.