joel/freqsplit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove unused parameter

Browse Source
This commit is contained in:
Joel Mathew Thomas
2025-01-28 20:33:28 +05:30
parent 7a7c27caa3
commit 3914b2c5ba
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/separation/convtasnet_wrapper.py
+1 -2
View File
@@ -1,13 +1,12 @@
import torch import torch
from asteroid.models import ConvTasNet from asteroid.models import ConvTasNet
def separate(audio, sr, model_name='mpariente/ConvTasNet_WHAMR_enhsingle'): def separate(audio, model_name='mpariente/ConvTasNet_WHAMR_enhsingle'):
""" """
Separates audio into sources using a pretrained Asteroid model. Separates audio into sources using a pretrained Asteroid model.
Args: Args:
audio (numpy.ndarray): The audio time series (1D numpy array). audio (numpy.ndarray): The audio time series (1D numpy array).
sr (int): Sampling rate of the audio.
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_WHAMR_enhsingle'.
Returns: Returns: