Hugging Face's logo

Spaces:
SkalskiP
/
florence-sam
Running on Zero

App Files Community
8
SkalskiP commited on
Commit
d1212b2
•
1 Parent(s): 57700a8

update device to CUDA

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ EXAMPLES = [
26
  "https://media.roboflow.com/notebooks/examples/dog-4.jpeg"
27
  ]
28
 
29
- DEVICE = torch.device("cpu")
30
 
31
  FLORENCE_MODEL, FLORENCE_PROCESSOR = load_florence_model(device=DEVICE)
32
  SAM_MODEL = load_sam_model(device=DEVICE)
 
26
  "https://media.roboflow.com/notebooks/examples/dog-4.jpeg"
27
  ]
28
 
29
+ DEVICE = torch.device("cuda")
30
 
31
  FLORENCE_MODEL, FLORENCE_PROCESSOR = load_florence_model(device=DEVICE)
32
  SAM_MODEL = load_sam_model(device=DEVICE)