change in model config

config.json

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

test.py

@@ -12,8 +12,15 @@
 
 # print(model)
 
-from transformers import TFAutoModel
+from transformers import AutoConfig, AutoModel
+from custom_model import SpleeterModel
 
-# Loading the TensorFlow model
-model = TFAutoModel.from_pretrained("niral-env/youtube_spleeter")
-print(model)
+# Load configuration and model
+config = AutoConfig.from_pretrained("<your-username>/<model-repo>")
+model = SpleeterModel(config)
+
+# Process an audio file
+# result = model.forward("example.wav")
+
+# Output separated stems
+print(model)