change model name

README.md

@@ -18,7 +18,7 @@ model-index:
 - name: Spleeter 2-Stems Transformer Integration
   results:
   - task:
-      type: fsmt
+      type: spleeter
       name: Source Separation
     dataset:
       type: None  # Replace with dataset name if you used one

custom_model.py

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