Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
2ec713c
1
Parent(s):
b7070f2
Fix GPU decorator usage in run_inference_gpu function
Browse files
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)
|