Spaces:

labofsahil
/

image

Runtime error

notsahil commited on Mar 25

Commit

7baed24

verified ·

1 Parent(s): c256298

Only detect person

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import PIL.Image as Image
 from ultralytics import ASSETS, YOLO
-model = YOLO("yolo11n.pt")
 def predict_image(img, conf_threshold, iou_threshold):
@@ -15,6 +15,7 @@ def predict_image(img, conf_threshold, iou_threshold):
         show_labels=True,
         show_conf=True,
         imgsz=640,
     )
     for r in results:

 from ultralytics import ASSETS, YOLO
+model = YOLO("yolo12n.pt")
 def predict_image(img, conf_threshold, iou_threshold):
         show_labels=True,
         show_conf=True,
         imgsz=640,
+        classes=[0]
     )
     for r in results: