prithivMLmods commited on
Commit
f933462
·
verified ·
1 Parent(s): 8a9b760

update app

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -130,9 +130,7 @@ def annotate_image(image: Image.Image, result: dict):
130
 
131
  points_array = np.array(points_list).reshape(1, -1, 2)
132
  key_points = sv.KeyPoints(xy=points_array)
133
- # Use bright red color (FF0000) for points
134
- bright_red = sv.Color(255, 0, 0) # RGB for bright red
135
- vertex_annotator = sv.VertexAnnotator(radius=6, color=bright_red) # Increased radius for better visibility
136
  annotated_image = vertex_annotator.annotate(scene=np.array(image.copy()), key_points=key_points)
137
  return Image.fromarray(annotated_image)
138
 
 
130
 
131
  points_array = np.array(points_list).reshape(1, -1, 2)
132
  key_points = sv.KeyPoints(xy=points_array)
133
+ vertex_annotator = sv.VertexAnnotator(radius=4, color=sv.Color.RED)
 
 
134
  annotated_image = vertex_annotator.annotate(scene=np.array(image.copy()), key_points=key_points)
135
  return Image.fromarray(annotated_image)
136