Hugging Face's logo

Spaces:
PSLD
/
PSLD
Sleeping

App Files Community
2
LituRout commited on
Commit
ab082bc
•
1 Parent(s): b105d3b

add gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ from transformers import AutoFeatureExtractor
30
  ## lr
31
  import torchvision
32
  import pdb
33
- os.environ['CUDA_VISIBLE_DEVICES']='1'
34
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
35
  import subprocess
36
  ##
 
30
  ## lr
31
  import torchvision
32
  import pdb
33
+ # os.environ['CUDA_VISIBLE_DEVICES']='1'
34
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
35
  import subprocess
36
  ##