Update app.py

app.py

@@ -12,9 +12,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def separate_audio(mixture):
     # Convert mixture to tensor
     print(mixture)
-    encoder_checkpoint = "models/encoder.pth"
-    decoder_checkpoint = "models/decoder.pth"
-    masknet_checkpoint = "models/masknet.pth"
+    encoder_checkpoint = "encoder.pth"
+    decoder_checkpoint = "decoder.pth"
+    masknet_checkpoint = "masknet.pth"
 
     encoder = torch.load(encoder_checkpoint, map_location=device)
     decoder = torch.load(decoder_checkpoint, map_location=device)
@@ -24,7 +24,7 @@ def separate_audio(mixture):
     # Step 2: Load Hyperparameters
 
     data_folder = "."
-    # hparams_file = "data/yamls/sepformer-customdataset.yaml"
+    # hparams_file = "sepformer-customdataset.yaml"
     overrides = f"data_folder: {data_folder}\noutput_folder: "
     hyperparams_file = "yamls/sepformer-customdataset.yaml"
     with open(hyperparams_file, "r") as f: