Spaces:
Sleeping
Sleeping
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)
|