ASesYusuf1 commited on
Commit
feb100b
·
verified ·
1 Parent(s): 6c32ddc

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.