From 5b84d3e3acf67eccf59dbf3c3e774bcd1eab4973 Mon Sep 17 00:00:00 2001
From: Elyas Heidari <55977725+EliHei2@users.noreply.github.com>
Date: Wed, 11 Dec 2024 07:22:03 +0100
Subject: [PATCH] Update __init__.py

---
 src/segger/prediction/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/segger/prediction/__init__.py b/src/segger/prediction/__init__.py
index 4539c8b..a98e0aa 100644
--- a/src/segger/prediction/__init__.py
+++ b/src/segger/prediction/__init__.py
@@ -4,6 +4,6 @@
 Contains prediction scripts and utilities for the Segger model.
 """
 
-__all__ = ["load_model", "predict"]
+__all__ = ["load_model", "segment"]
 
-from .predict import load_model, predict_parquet
+from .predict_parquet import load_model, segmenet