Spaces:
Sleeping
Sleeping
Commit
·
0a09e6c
1
Parent(s):
f634cd6
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
|