Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -990,7 +990,7 @@ def predict_with_paper(image, paper_size, offset, offset_unit, finger_clearance=
|
|
990 |
else:
|
991 |
# Set prompts for tool/object detection
|
992 |
yolo_world.set_classes(["tool", "object on paper"])
|
993 |
-
results = yolo_world.predict(masked_input_image, conf=0.
|
994 |
|
995 |
if not results or len(results) == 0 or not hasattr(results[0], 'boxes') or len(results[0].boxes) == 0:
|
996 |
logger.warning("No objects detected by YOLOWorld, proceeding with full image")
|
|
|
990 |
else:
|
991 |
# Set prompts for tool/object detection
|
992 |
yolo_world.set_classes(["tool", "object on paper"])
|
993 |
+
results = yolo_world.predict(masked_input_image, conf=0.2, verbose=False)
|
994 |
|
995 |
if not results or len(results) == 0 or not hasattr(results[0], 'boxes') or len(results[0].boxes) == 0:
|
996 |
logger.warning("No objects detected by YOLOWorld, proceeding with full image")
|