Update app.py

app.py

@@ -9,6 +9,9 @@ import torchaudio
 from speechbrain.lobes.models.dual_path import Encoder, SBTransformerBlock, SBTransformerBlock, Dual_Path_Model, Decoder
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
+sample_rate = 8000
+num_spks = 2
+ckpt_path = 'ckpts/'
 
 def load_sepformer(ckpt_path):
     encoder = Encoder(
@@ -35,7 +38,7 @@ def load_sepformer(ckpt_path):
         norm_before=True,
     )
     masknet = Dual_Path_Model(
-        num_spks=args.num_spks,
+        num_spks=num_spks,
         in_channels=256,
         out_channels=256,
         num_layers=2,
@@ -75,11 +78,9 @@ def load_fastgeco(ckpt_path):
     model.to(device)
     return model
 
-ckpt_path = 'ckpts/'
 encoder, masknet, decoder = load_sepformer(ckpt_path)
 fastgeco_model = load_fastgeco(ckpt_path)
-sample_rate = 8000
-num_spks = 2
+
 
 @spaces.GPU
 def separate(test_file, encoder, masknet, decoder):