models/base.py

from abc import ABC, abstractmethod
from typing import Optional

class BaseModelWrapper(ABC):
    @abstractmethod
    def load(self):
        pass

    @abstractmethod
    # Ensure segmentation_content is added here
    def predict(self, file_content: bytes, filename: str, audio_content: Optional[bytes] = None, segmentation_content: Optional[bytes] = None) -> dict:
        pass