linhcuem commited on
Commit
daf6e18
·
1 Parent(s): 46bfc84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -85,7 +85,8 @@ def yolov8_img_inference(
85
  # cls.add(model.names[int(cno)])
86
 
87
  # count_result = results.pandas().xyxy[0].value_counts('name')
88
- return render, present_objects
 
89
  # results = model.predict(image, imgsz=image_size, return_outputs=True)
90
  # results = model.predict(image)
91
  # object_prediction_list = []
 
85
  # cls.add(model.names[int(cno)])
86
 
87
  # count_result = results.pandas().xyxy[0].value_counts('name')
88
+ # return render, present_objects
89
+ return render, {names[k]: v for k, v in present_objects.items()}
90
  # results = model.predict(image, imgsz=image_size, return_outputs=True)
91
  # results = model.predict(image)
92
  # object_prediction_list = []