Hugging Face's logo

Spaces:
ai-for-good-lab
/
pytorch-wildlife
Running on CPU Upgrade

App Files Community
AndresHdzC commited on
Commit
d738bfa
·
verified ·
1 Parent(s): b38382a

Changed from cuda to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ from PytorchWildlife import utils as pw_utils
23
 
24
  #%%
25
  # Setting the device to use for computations ('cuda' indicates GPU)
26
- DEVICE = "cuda"
27
  # Initializing a supervision box annotator for visualizing detections
28
  box_annotator = sv.BoxAnnotator(thickness=4, text_thickness=4, text_scale=2)
29
 
 
23
 
24
  #%%
25
  # Setting the device to use for computations ('cuda' indicates GPU)
26
+ DEVICE = "cpu"
27
  # Initializing a supervision box annotator for visualizing detections
28
  box_annotator = sv.BoxAnnotator(thickness=4, text_thickness=4, text_scale=2)
29