Catmeow commited on
Commit
bed362e
1 Parent(s): 0c94286

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def inference(im):
10
  results._run(render=True)
11
  text = results.pandas().xyxy[0].round(2)
12
  counts = text.groupby(['name'])['name'].count()
13
- return Image.fromarray(results.ims[0]),str(counts)[4:-24]
14
 
15
  title = "Count Objects in the picture"
16
  description = "Count objects in picture by Yolov5s model"
 
10
  results._run(render=True)
11
  text = results.pandas().xyxy[0].round(2)
12
  counts = text.groupby(['name'])['name'].count()
13
+ return Image.fromarray(results.ims[0]),str(counts)[5:-24]
14
 
15
  title = "Count Objects in the picture"
16
  description = "Count objects in picture by Yolov5s model"