Update server.py

server.py

@@ -10,17 +10,13 @@ os.makedirs('uploaded', exist_ok=True)
 os.makedirs('processed', exist_ok=True)
 
 app = FastAPI()
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+model = build_audiosep(config_yaml='config/audiosep_base.yaml',checkpoint_path='checkpoint/audiosep_base_4M_steps.ckpt',device=device)
 
 # Mount the static files directory
 app.mount("/processed", StaticFiles(directory="processed"), name="processed")
 
 def process_audio(audio_file_path, text):
-    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-
-    model = build_audiosep(
-        config_yaml='config/audiosep_base.yaml',
-        checkpoint_path='checkpoint/audiosep_base_4M_steps.ckpt',
-        device=device)
 
     output_file = os.path.join('processed', 'SeparatedAudio.wav')
     separate_audio(model, audio_file_path, text, output_file, device)