Update app.py

app.py

@@ -242,23 +242,6 @@ def shrink_bbox(image: np.ndarray, shrink_factor: float):
     return cropped_image
 
 
-# def to_dxf(outlines):
-#     upper_range_tuple = (200)
-#     lower_range_tuple = (0)
-
-#     doc = ezdxf.new('R2010')
-#     msp = doc.modelspace()
-#     masked_jpg = cv2.inRange(outlines,lower_range_tuple, upper_range_tuple)
-
-#     for i in range(0,masked_jpg.shape[0]):
-#         for j in range(0,masked_jpg.shape[1]):
-#             if masked_jpg[i][j] == 255:
-#                 msp.add_line((j,masked_jpg.shape[0] - i), (j,masked_jpg.shape[0] - i))
-
-#     doc.saveas("./outputs/out.dxf")
-#     return "./outputs/out.dxf"
-
-
 def to_dxf(contours):
     doc = ezdxf.new()
     msp = doc.modelspace()
@@ -374,7 +357,7 @@ def predict(image, offset_inches):
     Image.fromarray(dilated_mask).save("./outputs/scaled_mask_new.jpg")
     outlines, contours = extract_outlines(dilated_mask)
     shrunked_img_contours = cv2.drawContours(
-        shrunked_img, contours, -1, (0, 0, 255), thickness=2
+        image, contours, -1, (0, 0, 255), thickness=2
     )
     dxf = save_dxf_spline(contours, scaling_factor, processed_size[0])