Hugging Face's logo

Spaces:
reab5555
/
Owlv2-Video-Object-Detection
Paused

App Files Community
reab5555 commited on
Commit
3f98689
·
verified ·
1 Parent(s): 05f5d03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import tempfile
11
  import shutil
12
 
13
  # Check if CUDA is available, otherwise use CPU
14
- device = "cuda"
15
 
16
  processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
17
  model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16")
 
11
  import shutil
12
 
13
  # Check if CUDA is available, otherwise use CPU
14
+ device = "cpu"
15
 
16
  processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
17
  model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16")