add model in predict.py

predict.py

 from test import CopyPredictor, ScaledMixturePredictor
 from test_umx import UMXPredictor
 from test_xumx import XUMXPredictor
+from test_xumx_2 import XUMXPredictor as MWFPredictor
 
 # Predictor which does nothing
 copy_predictor = CopyPredictor()
@@ -14,9 +15,11 @@ umx_predictor = UMXPredictor()
 # X-UMX needs `models` folder to be present in your submission, check test_xumx.py to learn more
 xumx_predictor = XUMXPredictor()
 
+mwf_predictor = MWFPredictor()
+
 """
 PARTICIPANT_TODO: The implementation you want to submit as your submission
 """
-submission = scaledmixture_predictor
+submission = mwf_predictor
 submission.run()
 print("Successfully completed music demixing...")

test_xumx_2.py

@@ -111,7 +111,7 @@ def separate(
 class XUMXPredictor(MusicDemixingPredictor):
     def prediction_setup(self):
         # Load your model here and put it into `evaluation` mode
-        self.separator = torch.jit.load("test_model.pt")
+        self.separator = torch.jit.load("models/test_model.pt")
         self.separator.eval()
 
     def prediction(