Update app.py

app.py

@@ -8,6 +8,7 @@ import tempfile
 import logging
 from datetime import datetime
 from typing import Tuple, List, Optional
+import space
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -42,6 +43,7 @@ class MusicRemover:
         return samples, sample_rate, audio
     
     @torch.no_grad()
+    @spaces.GPU()
     def detect_music_segments(self, audio_array, sample_rate, threshold, window_size, hop_size):
         window_samples = int(window_size * sample_rate)
         hop_samples = int(hop_size * sample_rate)