Spaces:

Catmeow
/

Count_objects_in_picture

Runtime error

Catmeow commited on Nov 3, 2022

Commit

0c94286

•

1 Parent(s): d1e3482

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def inference(im):
     results._run(render=True)
     text = results.pandas().xyxy[0].round(2)
     counts = text.groupby(['name'])['name'].count()
-    return Image.fromarray(results.ims[0]),str(counts)[4:-20]
 title = "Count Objects in the picture"
 description = "Count objects in picture by Yolov5s model"

     results._run(render=True)
     text = results.pandas().xyxy[0].round(2)
     counts = text.groupby(['name'])['name'].count()
+    return Image.fromarray(results.ims[0]),str(counts)[4:-24]
 title = "Count Objects in the picture"
 description = "Count objects in picture by Yolov5s model"