mlbench123 commited on
Commit
4ce1663
·
verified ·
1 Parent(s): 6321dbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.5, 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")
 
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")