Hugging Face's logo

Spaces:
pg56714
/
YoloWGDinoArena
Sleeping

App Files Community
pg56714 commited on
Commit
667bd92
•
1 Parent(s): 6989469

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -127,7 +127,6 @@ def process_image(
127
  ) -> np.ndarray:
128
  categories = process_categories(categories)
129
  YOLO_WORLD_MODEL.set_classes(categories)
130
- YOLO_WORLD_MODEL.set_classes(categories)
131
  results = YOLO_WORLD_MODEL.infer(input_image, confidence=confidence_threshold)
132
  detections = sv.Detections.from_inference(results).with_nms(
133
  class_agnostic=True, threshold=nms_threshold
 
127
  ) -> np.ndarray:
128
  categories = process_categories(categories)
129
  YOLO_WORLD_MODEL.set_classes(categories)
 
130
  results = YOLO_WORLD_MODEL.infer(input_image, confidence=confidence_threshold)
131
  detections = sv.Detections.from_inference(results).with_nms(
132
  class_agnostic=True, threshold=nms_threshold