PawelMar commited on
Commit
c10751c
1 Parent(s): d987f60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,10 +7,11 @@ def yoloV8_func(image: gr.Image = None):
7
  model_path = "models/last.pt"
8
  model = YOLO(model_path)
9
  results = model.predict(image)
 
10
  for result in results:
11
  if result.masks:
12
- masks_json = [{"x": format(point[0], '.2f'), "y": format(point[1], '.2f')} for point in result.masks.xy[0]]
13
- json.dumps(masks_json)
14
  return masks_json
15
 
16
 
 
7
  model_path = "models/last.pt"
8
  model = YOLO(model_path)
9
  results = model.predict(image)
10
+ masks_json = []
11
  for result in results:
12
  if result.masks:
13
+ masks_json.append([{"x": format(point[0], '.2f'), "y": format(point[1], '.2f')} for point in result.masks.xy[0]])
14
+ masks_json = json.dumps(masks_json)
15
  return masks_json
16
 
17