wickedreg commited on
Commit
42ebd1b
·
verified ·
1 Parent(s): 6469dcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -29,14 +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
-
33
- detected_fruit = None
34
-
35
- for det in detections:
36
- label = model.names[int(det.boxes.cls[0])] # название фрукта
37
- if label in fruits_data:
38
- detected_fruit = label
39
- break
40
 
41
  return result_np_image, detected_fruit
42
 
 
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