From 88ebfb66f652d73e98767638ab1ce8b0bba6fcd3 Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Tue, 28 Jan 2025 22:39:45 +0530
Subject: [PATCH] change model

---
 src/separation/convtasnet_wrapper.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/separation/convtasnet_wrapper.py b/src/separation/convtasnet_wrapper.py
index 0c08b2a..a277762 100644
--- a/src/separation/convtasnet_wrapper.py
+++ b/src/separation/convtasnet_wrapper.py
@@ -1,13 +1,13 @@
 import torch
 from asteroid.models import ConvTasNet
 
-def separate(audio, model_name='mpariente/ConvTasNet_WHAMR_enhsingle'):
+def separate(audio, model_name='mpariente/ConvTasNet_WHAM!_sepclean'):
     """
     Separates audio into sources using a pretrained Asteroid model.
 
     Args:
         audio (numpy.ndarray): The audio time series (1D numpy array).
-        model_name (str): Name of the pretrained model from Asteroid. Default is 'mpariente/ConvTasNet_WHAMR_enhsingle'.
+        model_name (str): Name of the pretrained model from Asteroid. Default is 'mpariente/ConvTasNet_WHAM!_sepclean'.
 
     Returns:
         list: List of separated sources as numpy arrays.