Update audio_separator/separator/separator.py

audio_separator/separator/separator.py

@@ -392,6 +392,7 @@ class Separator:
             audio_data = audio_data * (self.amplification_threshold / max_amplitude)
         return torch.tensor(audio_data, dtype=torch.float32, device=self.torch_device)
 
+    @spaces.GPU
     def separate(self, audio_file_path, custom_output_names=None):
         """Separate audio file into stems."""
         if not self.model_instance:
@@ -418,6 +419,7 @@ class Separator:
         self.logger.info(f"Separation completed in {time.perf_counter() - start_time:.2f} seconds")
         return output_files
 
+    @spaces.GPU
     def _separate_file(self, audio_file_path, custom_output_names=None):
         """Internal method to separate a single audio file."""
         self.logger.debug(f"Processing file: {audio_file_path}")