Update ensemble.py

ensemble.py

@@ -6,6 +6,7 @@ import librosa
 import soundfile as sf
 import numpy as np
 import argparse
+import spaces
 
 def stft(wave, nfft, hl):
     wave_left = np.asfortranarray(wave[0])
@@ -111,6 +112,7 @@ def average_waveforms(pred_track, weights, algorithm):
         pred_track = istft(pred_track, 1024, final_length)
     return pred_track
 
+@spaces.GPU
 def ensemble_files(args):
     parser = argparse.ArgumentParser()
     parser.add_argument("--files", type=str, required=True, nargs='+', help="Path to all audio-files to ensemble")