wickedreg commited on
Commit
03e0a23
·
verified ·
1 Parent(s): 5a31774

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -33,11 +33,10 @@ def detect_fruit(image):
33
  detected_fruit = None
34
 
35
  for det in detections:
36
- print (det)
37
- # label = model.names[int(det[5])] # название фрукта
38
- # if label in fruits_data:
39
- # detected_fruit = label
40
- # break
41
 
42
  return result_np_image, detected_fruit
43
 
 
33
  detected_fruit = None
34
 
35
  for det in detections:
36
+ label = model.names[9] # название фрукта
37
+ if label in fruits_data:
38
+ detected_fruit = label
39
+ break
 
40
 
41
  return result_np_image, detected_fruit
42