diff --git a/model_api/src/model_api/models/__init__.py b/model_api/src/model_api/models/__init__.py
index 0d4e62f4..ad83f3be 100644
--- a/model_api/src/model_api/models/__init__.py
+++ b/model_api/src/model_api/models/__init__.py
@@ -19,6 +19,7 @@
     DetectionResult,
     ImageResultWithSoftPrediction,
     InstanceSegmentationResult,
+    Label,
     PredictedMask,
     RotatedSegmentationResult,
     VisualPromptingResult,
@@ -89,7 +90,6 @@
     "SAMImageEncoder",
     "SAMLearnableVisualPrompter",
     "SAMVisualPrompter",
-    "SalientObjectDetectionModel",
     "segmentation_models",
     "SegmentationModel",
     "SSD",
@@ -100,8 +100,6 @@
     "YOLODETR",
     "YOLOSeg",
     "YOLOF",
-    "YOLOv3ONNX",
-    "YOLOv4",
     "YOLOv5",
     "YOLOv8",
     "YOLOX",