Spaces:
Running
on
Zero
Running
on
Zero
Update audio_separator/separator/separator.py
Browse files
audio_separator/separator/separator.py
CHANGED
@@ -20,7 +20,6 @@ import torch
|
|
20 |
import torch.amp.autocast_mode as autocast_mode
|
21 |
import onnxruntime as ort
|
22 |
from tqdm import tqdm
|
23 |
-
import spaces
|
24 |
|
25 |
|
26 |
class Separator:
|
@@ -778,7 +777,6 @@ class Separator:
|
|
778 |
self.logger.debug("Loading model completed.")
|
779 |
self.logger.info(f'Load model duration: {time.strftime("%H:%M:%S", time.gmtime(int(time.perf_counter() - load_model_start_time)))}')
|
780 |
|
781 |
-
@spaces.GPU
|
782 |
def separate(self, audio_file_path, custom_output_names=None):
|
783 |
"""
|
784 |
Separates the audio file(s) into different stems (e.g., vocals, instruments) using the loaded model.
|
@@ -832,7 +830,6 @@ class Separator:
|
|
832 |
|
833 |
return output_files
|
834 |
|
835 |
-
@spaces.GPU
|
836 |
def _separate_file(self, audio_file_path, custom_output_names=None):
|
837 |
"""
|
838 |
Internal method to handle separation for a single audio file.
|
|
|
20 |
import torch.amp.autocast_mode as autocast_mode
|
21 |
import onnxruntime as ort
|
22 |
from tqdm import tqdm
|
|
|
23 |
|
24 |
|
25 |
class Separator:
|
|
|
777 |
self.logger.debug("Loading model completed.")
|
778 |
self.logger.info(f'Load model duration: {time.strftime("%H:%M:%S", time.gmtime(int(time.perf_counter() - load_model_start_time)))}')
|
779 |
|
|
|
780 |
def separate(self, audio_file_path, custom_output_names=None):
|
781 |
"""
|
782 |
Separates the audio file(s) into different stems (e.g., vocals, instruments) using the loaded model.
|
|
|
830 |
|
831 |
return output_files
|
832 |
|
|
|
833 |
def _separate_file(self, audio_file_path, custom_output_names=None):
|
834 |
"""
|
835 |
Internal method to handle separation for a single audio file.
|