Nathan Pruyne commited on
Commit
b63dc56
·
1 Parent(s): 8a8ea06

Add ZeroGPU capability

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from timbre_trap.framework.modules import TimbreTrap
2
  from pyharp import *
3
 
@@ -32,7 +33,7 @@ model_card = ModelCard(
32
  tags=['example', 'music transcription', 'multi-pitch estimation', 'timbre filtering']
33
  )
34
 
35
-
36
  def process_fn(audio_path, transcribe):#, demo):
37
  # Load the audio with torchaudio
38
  audio, fs = torchaudio.load(audio_path)
 
1
+ import spaces
2
  from timbre_trap.framework.modules import TimbreTrap
3
  from pyharp import *
4
 
 
33
  tags=['example', 'music transcription', 'multi-pitch estimation', 'timbre filtering']
34
  )
35
 
36
+ @spaces.GPU
37
  def process_fn(audio_path, transcribe):#, demo):
38
  # Load the audio with torchaudio
39
  audio, fs = torchaudio.load(audio_path)