linhcuem commited on
Commit
119cb8d
1 Parent(s): 840712d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -59,7 +59,7 @@ def yolov8_img_inference(
59
 
60
  for r in results:
61
  for c in r.boxes.cls:
62
- print(names[int(c)])
63
 
64
  # clist= results[0].boxes.cls.tolist()
65
  # cls = set()
@@ -67,8 +67,7 @@ def yolov8_img_inference(
67
  # cls.add(model.names[int(cno)])
68
 
69
  # count_result = results.pandas().xyxy[0].value_counts('name')
70
- return render, print(names[int(c)])
71
-
72
  # results = model.predict(image, imgsz=image_size, return_outputs=True)
73
  # results = model.predict(image)
74
  # object_prediction_list = []
 
59
 
60
  for r in results:
61
  for c in r.boxes.cls:
62
+ c = print(names[int(c)])
63
 
64
  # clist= results[0].boxes.cls.tolist()
65
  # cls = set()
 
67
  # cls.add(model.names[int(cno)])
68
 
69
  # count_result = results.pandas().xyxy[0].value_counts('name')
70
+ return render, c
 
71
  # results = model.predict(image, imgsz=image_size, return_outputs=True)
72
  # results = model.predict(image)
73
  # object_prediction_list = []