rollback to use lfs

test_unet.py

@@ -102,11 +102,7 @@ def separate(
 class UNetAttnPredictor(MusicDemixingPredictor):
     def prediction_setup(self):
         # Load your model here and put it into `evaluation` mode
-        model_path, _ = download_large_file_from_google_drive(
-            "1GemMK3ETH5jJV-hnFfAyRCGrN2gDe5bo",
-            save_path="./models", save_name="unet.pth"
-        )
-        self.separator = torch.jit.load(model_path)
+        self.separator = torch.jit.load('./models/unet_test.pth')
         self.separator.eval()
 
     def prediction(