Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
210 |
-
TEXT_THRESHOLD = 0.
|
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
|