Spaces:
Sleeping
Sleeping
Commit
·
f2d9d9a
1
Parent(s):
7eb36e4
Application file
Browse files
app.py
CHANGED
|
@@ -37,11 +37,6 @@ processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
|
|
| 37 |
model_sam = SamModel.from_pretrained("facebook/sam-vit-huge").to(device)
|
| 38 |
processor_sam = SamProcessor.from_pretrained("facebook/sam-vit-huge")
|
| 39 |
|
| 40 |
-
# model = Owlv2ForObjectDetection.from_pretrained("owlv2-base-patch16-ensemble").to(device)
|
| 41 |
-
# processor = Owlv2Processor.from_pretrained("owlv2-base-patch16-ensemble")
|
| 42 |
-
# model_sam = SamModel.from_pretrained("SAM/sam-vit-huge").to(device)
|
| 43 |
-
# processor_sam = SamProcessor.from_pretrained("SAM/sam-vit-huge")
|
| 44 |
-
|
| 45 |
|
| 46 |
@spaces.GPU
|
| 47 |
def query_image(img, text_queries, score_threshold=0.5):
|
|
|
|
| 37 |
model_sam = SamModel.from_pretrained("facebook/sam-vit-huge").to(device)
|
| 38 |
processor_sam = SamProcessor.from_pretrained("facebook/sam-vit-huge")
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
|
| 41 |
@spaces.GPU
|
| 42 |
def query_image(img, text_queries, score_threshold=0.5):
|