35 lines
818 B
Python
35 lines
818 B
Python
import logging
|
|
from .scripts.vocal_and_sound_remover import VocalAndSoundRemover
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class VocalAndSoundRemoverNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input_sound": ("AUDIO",),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("AUDIO","AUDIO",)
|
|
RETURN_NAMES = ("Vocals","Background",)
|
|
FUNCTION = "execute"
|
|
CATEGORY = "DeepExtract"
|
|
|
|
def execute(self, input_sound):
|
|
vocal_remover = VocalAndSoundRemover(input_sound)
|
|
|
|
vocals, background = vocal_remover.execute()
|
|
|
|
return (vocals, background,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"VocalAndSoundRemoverNode": VocalAndSoundRemoverNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"VocalAndSoundRemoverNode": "Vocal and Sound Separator",
|
|
}
|