tfmoraes

diff --git a/invesalius/segmentation/deep_learning/segment.py b/invesalius/segmentation/deep_learning/segment.py
index 041e6f90..dc17a203 100644
--- a/invesalius/segmentation/deep_learning/segment.py
+++ b/invesalius/segmentation/deep_learning/segment.py
@@ -5,8 +5,13 @@ import pathlib
 import sys
 import tempfile
 import traceback
+from typing import Generator, Tuple
 

tfmoraes

Mar 25 10:22:24 bluefin kernel: Linux version 6.14.0-0.rc7.56.fc42.x86_64 (mockbuild@f8900533f8d8482aaf6e1e5063eccb1a) (gcc (GCC) 15.0.1 20250228 (Red Hat 15.0.1-0), GNU ld version 2.44-3.fc42) #1 SMP PREEMPT_DYNAMIC Mon Mar 17 13:25:24 UTC 2025
Mar 25 10:22:24 bluefin kernel: Command line: BOOT_IMAGE=(hd1,gpt2)/ostree/default-b977a7f3000d6a9c9d29f7e2a83a0471a543f9309a719a40e737e784d4b35f44/vmlinuz-6.14.0-0.rc7.56.fc42.x86_64 rd.luks.uuid=luks-230af0e7-091c-4f95-ab6c-dfc1c1dc61b7 rhgb quiet root=UUID=29051c3c-4953-45af-ae96-c4d3e51f67bf rootflags=subvol=root rw ostree=/ostree/boot.1/default/b977a7f3000d6a9c9d29f7e2a83a0471a543f9309a719a40e737e784d4b35f44/0 initcall_blacklist=simpledrm_platform_driver_init
Mar 25 10:22:24 bluefin kernel: BIOS-provided physical RAM map:
Mar 25 10:22:24 bluefin kernel: BIOS-e820: [mem 0x0000000000000000-0x000000000009ffff] usable
Mar 25 10:22:24 bluefin kernel: BIOS-e820: [mem 0x00000000000a0000-0x00000000000fffff] reserved
Mar 25 10:22:24 bluefin kernel: BIOS-e820: [mem 0x00000