Spaces:
Paused
Paused
Added cuda
Browse files
app.py
CHANGED
|
@@ -4,7 +4,9 @@ from flask.Genre_spotting_service import _Genre_spotting_service
|
|
| 4 |
from flask.Beat_tracking_service import _Beat_tracking_service
|
| 5 |
from diffusers import StableDiffusionPipeline
|
| 6 |
import torch
|
|
|
|
| 7 |
|
|
|
|
| 8 |
emo_list = []
|
| 9 |
gen_list = []
|
| 10 |
tempo_list = []
|
|
|
|
| 4 |
from flask.Beat_tracking_service import _Beat_tracking_service
|
| 5 |
from diffusers import StableDiffusionPipeline
|
| 6 |
import torch
|
| 7 |
+
import os
|
| 8 |
|
| 9 |
+
os.environ['CUDA_VISIBLE_DEVICES'] = '4'
|
| 10 |
emo_list = []
|
| 11 |
gen_list = []
|
| 12 |
tempo_list = []
|