Saad0KH commited on
Commit
6801b2a
·
verified ·
1 Parent(s): d64d60f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -78,6 +78,9 @@ detector.prepare(-1, nms_thresh=0.5, input_size=(640, 640))
78
 
79
 
80
  def detect(image: np.ndarray) -> tuple[Image.Image, list[Image.Image]]:
 
 
 
81
  image = image[:, :, ::-1] # RGB -> BGR
82
  bboxes, vbboxes = detect_person(image, detector)
83
  res = visualize(image, bboxes, vbboxes)
 
78
 
79
 
80
  def detect(image: np.ndarray) -> tuple[Image.Image, list[Image.Image]]:
81
+ if image is None:
82
+ return None, []
83
+
84
  image = image[:, :, ::-1] # RGB -> BGR
85
  bboxes, vbboxes = detect_person(image, detector)
86
  res = visualize(image, bboxes, vbboxes)