JacobLinCool commited on
Commit
2ec713c
·
1 Parent(s): b7070f2

Fix GPU decorator usage in run_inference_gpu function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -226,7 +226,7 @@ def infer_tc7(audio, nps, bpm, offset, difficulty, level, DEVICE, MODEL):
226
  return oni_audio, plot, tja_content
227
 
228
 
229
- @spaces.GPU
230
  def run_inference_gpu(audio, model_choice, nps, bpm, offset, difficulty, level):
231
  if model_choice == "TC5":
232
  return infer_tc5(audio, nps, bpm, offset, GPU_DEVICE, tc5)
 
226
  return oni_audio, plot, tja_content
227
 
228
 
229
+ @spaces.GPU()
230
  def run_inference_gpu(audio, model_choice, nps, bpm, offset, difficulty, level):
231
  if model_choice == "TC5":
232
  return infer_tc5(audio, nps, bpm, offset, GPU_DEVICE, tc5)