wickedreg commited on
Commit
6469dcf
·
verified ·
1 Parent(s): 0cdff7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def detect_fruit(image):
24
  detections = model.predict(source=image_cv, conf=0.5)
25
 
26
  # проверка на наличие детекций
27
- if len(results[0].boxes) == 0:
28
  return image_cv, None
29
 
30
  result_np_image = detections[0].plot()
 
24
  detections = model.predict(source=image_cv, conf=0.5)
25
 
26
  # проверка на наличие детекций
27
+ if len(detections[0].boxes.cls) == 0:
28
  return image_cv, None
29
 
30
  result_np_image = detections[0].plot()