Update app.py

app.py

@@ -113,6 +113,7 @@ def postprocessor(prediction, orig_img):
     img_copy = copy.deepcopy(orig_img)
     result_image = None
     text = ["A", "B"]
+    points = list()
     for bbox in best_boxes:
         start_point = (int(bbox[0]), int(bbox[1]))
         end_point = (int(bbox[2]), int(bbox[3]))
@@ -124,10 +125,13 @@ def postprocessor(prediction, orig_img):
         for idx, point in enumerate(ab): 
             x = int(point[0])
             y = int(point[1])
+            points.append((x,y))
             color = (255, 0, 0)
-            thickness = 0.5
+            thickness = 1
             result_image = cv2.circle(result_image, (x,y), radius=0, color=color, thickness=thickness)
-            result_image = cv2.putText(result_image,text[idx], (x-10,y+10), cv2.FONT_HERSHEY_SIMPLEX, 1, (255,0,0), thickness, cv2.LINE_AA)
+            result_image = cv2.putText(result_image,text[idx], (x-10,y+10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255,0,0), thickness, cv2.LINE_AA)
+    
+    result_image = cv2.line(result_image, points[0], points[1], (255,0,0), thickness)
     return result_image