ChristopherMarais commited on
Commit
9b18251
·
verified ·
1 Parent(s): e65131a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -206,8 +206,8 @@ dinov2_transform = transforms.Compose([
206
  # Object Detection Function
207
  def detect_objects(og_image, model=od_model, prompt="bug . insect", device="cpu"):
208
  TEXT_PROMPT = prompt
209
- BOX_THRESHOLD = 0.35 # Adjusted back to original value
210
- TEXT_THRESHOLD = 0.25 # Adjusted back to original value
211
  DEVICE = device # cuda or cpu
212
 
213
  # Convert numpy array to PIL Image if needed
 
206
  # Object Detection Function
207
  def detect_objects(og_image, model=od_model, prompt="bug . insect", device="cpu"):
208
  TEXT_PROMPT = prompt
209
+ BOX_THRESHOLD = 0.15 # 35 Adjusted back to original value
210
+ TEXT_THRESHOLD = 0.15 # 25 Adjusted back to original value
211
  DEVICE = device # cuda or cpu
212
 
213
  # Convert numpy array to PIL Image if needed