From 3c284b4ab4a2cff12256dc62e595f742a89be406 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Anna=20B=C3=B8gevang=20Ekner?= <s193396@dtu.dk>
Date: Tue, 4 Feb 2025 14:36:23 +0100
Subject: [PATCH] updated segmentation pipeline notebook

---
 docs/notebooks/segmentation_pipeline.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/notebooks/segmentation_pipeline.ipynb b/docs/notebooks/segmentation_pipeline.ipynb
index 326ab9d7..13c986e9 100644
--- a/docs/notebooks/segmentation_pipeline.ipynb
+++ b/docs/notebooks/segmentation_pipeline.ipynb
@@ -528,7 +528,7 @@
    "outputs": [],
    "source": [
     "# defining model\n",
-    "model = qim3d.ml.models.UNet(size = 'small', dropout = 0.25)"
+    "model = qim3d.ml.models.UNet2D(size = 'small', dropout = 0.25)"
    ]
   },
   {
-- 
GitLab