removed unused object

core/database/utils.py

@@ -4,9 +4,9 @@ import logging
 
 from pymongo.collection import Collection
 
-from .classes import ModelOutputs
 from .classes import NoDocumentFoundException
 from .classes import VisualCommunication
+from core.dto import ModelData
 
 
 def count_documents(
@@ -97,7 +97,7 @@ def get_visual_communication(
 def upsert_predictions(
     collection: Collection,
     vis_com_name: str,
-    predictions: ModelOutputs,
+    predictions: ModelData,
 ) -> None:
     """Upsert prediction data in the database."""
     query = {
@@ -120,7 +120,7 @@ def upsert_predictions(
 def upsert_annotations(
     collection: Collection,
     vis_com_name: str,
-    annotations: ModelOutputs,
+    annotations: ModelData,
 ) -> None:
     """Upserts annotation data in the database."""
     query = {