Hugging Face's logo

Spaces:
jgitsolutions
/
zeroshot_object_tracking_OWLv2
Sleeping

App Files Community
merve HF staff commited on
Commit
a3c1650
·
verified ·
1 Parent(s): 554867f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from PIL import Image
13
  import spaces
14
 
15
 
16
- device = torch.device( cuda if torch.cuda.is_available() else cpu )
17
  processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
18
  model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
19
 
 
13
  import spaces
14
 
15
 
16
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
  processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
18
  model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
19