Update app.py

app.py

@@ -14,7 +14,7 @@ separator = torch.hub.load('sigsep/open-unmix-pytorch', 'umxhq')
 def inference(audio):
     audio, rate = stempeg.read_stems(
         audio,
-        sample_rate=44100
+        sample_rate=separator.sample_rate
     )
     audio = torch.as_tensor(audio).float().T
     audio = audio[None]
@@ -29,7 +29,7 @@ def inference(audio):
         target_path,
         estimates_numpy,
         sample_rate=rate,
-        writer=stempeg.FilesWriter(multiprocess=True, output_sample_rate=44100),
+        writer=stempeg.FilesWriter(multiprocess=True, output_sample_rate=separator.sample_rate),
     )
     return 'vocals.wav', 'drums.wav', 'bass.wav', 'other.wav'