change model name

config.json

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

custom_model.py

@@ -28,5 +28,5 @@ class SpleeterModel(PreTrainedModel):
         return self.separator.separate(audio_path)
 
 
-AutoConfig.register("fsmt", SpleeterConfig)  # Register the custom configuration
+AutoConfig.register("spleeter", SpleeterConfig)  # Register the custom configuration
 AutoModel.register(SpleeterConfig, SpleeterModel)  # Register the custom model

test.py

@@ -7,7 +7,13 @@
 
 
 # Load model directly
-from transformers import AutoModel
-model = AutoModel.from_pretrained("niral-env/youtube_spleeter")
+# from transformers import AutoModel
+# model = AutoModel.from_pretrained("niral-env/youtube_spleeter")
 
+# print(model)
+
+from transformers import TFAutoModel
+
+# Loading the TensorFlow model
+model = TFAutoModel.from_pretrained("niral-env/youtube_spleeter")
 print(model)