Spaces:

merve
/

owl-tracking

Running on Zero

merve HF staff commited on Feb 21

Commit

a3c1650

•

1 Parent(s): 554867f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from PIL import Image
 import spaces
-device = torch.device( cuda if torch.cuda.is_available() else cpu )
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)

 import spaces
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)