fix: demucs not working

Author: kerthical

backend/serve.py

@@ -10,8 +10,6 @@
 import torch
 import torch.nn.functional as F
 import torchaudio
-from demucs import pretrained
-from demucs.apply import apply_model
 from fastapi import HTTPException, UploadFile
 from model import DialectClassifier
 from starlette.middleware.cors import CORSMiddleware
@@ -31,7 +29,6 @@ def setup(self, device: str):
             map_location=device,
         )
         self.model.eval()
-        self.demucs_model = pretrained.get_model("htdemucs")
         self.transform = A.Compose(
             [
                 A.AddGaussianNoise(p=0.2),
@@ -54,13 +51,8 @@ def setup(self, device: str):
     def separate_audio(self, audio_path: str) -> str:
         wav, sr = torchaudio.load(audio_path)
         wav = wav.mean(dim=0, keepdim=True)
-
-        with torch.no_grad():
-            sources = apply_model(self.demucs_model, wav.unsqueeze(0), shifts=1)
-
-        vocals = sources[0, 3]
         separated_path = f"/tmp/separated_{time.time()}.wav"
-        torchaudio.save(separated_path, vocals, sr)
+        torchaudio.save(uri=separated_path, src=wav, sample_rate=sr)
         return separated_path
 
     def remove_silence(self, audio: np.ndarray, sr: int) -> np.ndarray: