Spaces:
Running
Running
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.
|