wickedreg commited on
Commit
95668d2
1 Parent(s): 84bc0ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def detect_fruit(image):
29
 
30
  result_np_image = detections[0].plot()
31
  result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
32
- detected_fruit = model.names[int(detections[].boxes.cls[0])]
33
 
34
  return result_np_image, detected_fruit
35
 
 
29
 
30
  result_np_image = detections[0].plot()
31
  result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
32
+ detected_fruit = model.names[int(detections[0].boxes.cls[0])]
33
 
34
  return result_np_image, detected_fruit
35