Add Transformers integration with custom model

.gitignore

@@ -0,0 +1 @@
+env/

config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "spleeter",
+    "stems": 2
+}

custom_model.py

@@ -0,0 +1,25 @@
+from transformers import PreTrainedModel, PretrainedConfig
+from spleeter.separator import Separator
+
+class SpleeterConfig(PretrainedConfig):
+    model_type = "spleeter"
+    def __init__(self, stems=2, **kwargs):
+        super().__init__(**kwargs)
+        self.stems = stems
+
+class SpleeterModel(PreTrainedModel):
+    config_class = SpleeterConfig
+
+    def __init__(self, config):
+        super().__init__(config)
+        self.separator = Separator(f"{config.stems}stems")
+
+    def forward(self, audio_path: str):
+        """
+        Separates the stems in the given audio file.
+        Args:
+            audio_path (str): Path to the input audio file.
+        Returns:
+            dict: Separated stems.
+        """
+        return self.separator.separate(audio_path)