Hugging Face's logo

Spaces:
merve
/
OWLSAM
Running on Zero

App Files Community
3
merve HF Staff commited on
Commit
026a244
·
verified ·
1 Parent(s): ee1bcff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  import numpy as np
4
  import spaces
5
 
6
- checkpoint = "google/owlvit-base-patch16"
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")
 
3
  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")