Spaces:

merve
/

OWLSAM

Running on Zero

merve HF staff commited on Jul 15

Commit

026a244

•

1 Parent(s): ee1bcff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 import numpy as np
 import spaces
-checkpoint = "google/owlvit-base-patch16"
 detector = pipeline(model=checkpoint, task="zero-shot-object-detection")
 sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")

 import numpy as np
 import spaces
+checkpoint = "google/owlv2-base-patch16-ensemble"
 detector = pipeline(model=checkpoint, task="zero-shot-object-detection")
 sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")