update u2net-lite model link in background removal notebook (#1711)

openvinotoolkit · Feb 14, 2024 · cd2251c · cd2251c
1 parent 898d8d4
commit cd2251c
Showing 1 changed file with 6 additions and 37 deletions.
diff --git a/notebooks/205-vision-background-removal/205-vision-background-removal.ipynb b/notebooks/205-vision-background-removal/205-vision-background-removal.ipynb
@@ -14,7 +14,8 @@
     "The PyTorch U$^2$-Net model is converted to OpenVINO IR format. The model source is available [here](https://github.com/xuebinqin/U-2-Net).\n",
     "\n",
     "\n",
-    "#### Table of contents:\n\n",
+    "#### Table of contents:\n",
+    "\n",
     "- [Preparation](#Preparation)\n",
     "    - [Install requirements](#Install-requirements)\n",
     "    - [Import the PyTorch Library and U$^2$-Net](#Import-the-PyTorch-Library-and-U2-Net)\n",
@@ -150,7 +151,7 @@
     "\n",
     "u2net_lite = model_config(\n",
     "    name=\"u2net_lite\",\n",
-    "    url=\"https://drive.google.com/uc?id=1rbSTGKAE-MTxBYHd-51l2hMOQPT_7EPy\",\n",
+    "    url=\"https://drive.google.com/uc?id=1W8E4FHIlTVstfRkYmNOjbr0VDXTZm0jD\",\n",
     "    model=U2NETP,\n",
     "    model_args=(),\n",
     ")\n",
@@ -162,7 +163,7 @@
     ")\n",
     "u2net_human_seg = model_config(\n",
     "    name=\"u2net_human_seg\",\n",
-    "    url=\"https://drive.google.com/uc?id=1-Yg0cxgrNhHP-016FPdp902BR-kSsA4P\",\n",
+    "    url=\"https://drive.google.com/uc?id=1m_Kgs91b21gayc2XLW0ou8yugAIadWVP\",\n",
     "    model=U2NET,\n",
     "    model_args=(3, 1),\n",
     ")\n",
@@ -701,44 +702,12 @@
   },
   "widgets": {
    "application/vnd.jupyter.widget-state+json": {
-    "state": {
-     "08172c8cd5e943cfb6a9e4e2ec33ab13": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "DropdownModel",
-      "state": {
-       "_options_labels": [
-        "CPU",
-        "GPU.0",
-        "GPU.1",
-        "AUTO"
-       ],
-       "description": "Device:",
-       "index": 3,
-       "layout": "IPY_MODEL_61f88e29badf43fe9ebabb3ecbc56e2a",
-       "style": "IPY_MODEL_b186e86e45b94c2ca7c09c2933ace135"
-      }
-     },
-     "61f88e29badf43fe9ebabb3ecbc56e2a": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {}
-     },
-     "b186e86e45b94c2ca7c09c2933ace135": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "DescriptionStyleModel",
-      "state": {
-       "description_width": ""
-      }
-     }
-    },
+    "state": {},
     "version_major": 2,
     "version_minor": 0
    }
   }
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}