Hugging Face's logo

Spaces:
merve
/
OWLSAM
Running on Zero

App Files Community
1
merve HF staff commited on
Commit
d4be87c
•
1 Parent(s): 47a12b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import spaces
5
 
6
  checkpoint = "google/owlv2-base-patch16-ensemble"
7
- detector = pipeline(model=checkpoint, task="zero-shot-object-detection")
8
  sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
9
  sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
10
 
 
4
  import spaces
5
 
6
  checkpoint = "google/owlv2-base-patch16-ensemble"
7
+ detector = pipeline(model=checkpoint, task="zero-shot-object-detection", device="cuda")
8
  sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
9
  sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
10