Update app.py

app.py

@@ -21,12 +21,11 @@ def midpoint(x1, y1, x2, y2):
     return (x_mid, y_mid)
 
 def draw_boxes(img, bounds, color='yellow', width=2):
-    
     mask = np.zeros(img.shape[:2], dtype="uint8")
-    draw = ImageDraw.Draw(mask)
     for bound in bounds:
-        p0, p1, p2, p3 = bound[0]
-        draw.polygon((*p0, *p1, *p2, *p3), fill=255)
+        pts = np.array([bound[0]], np.int32)
+        pts = pts.reshape((-1,1,2))
+        cv.polylines(mask,[pts],True,(0,255,255))
     return (mask)
     
 def draw_mask(img, bounds):
@@ -45,9 +44,9 @@ def draw_mask(img, bounds):
         thickness = int(math.sqrt((x2 - x1)**2 + (y2 - y1)**2))
         
         cv.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
-    img = cv.inpaint(img, mask, 1, cv.INPAINT_NS)
+    #img = cv.inpaint(img, mask, 1, cv.INPAINT_NS)
                  
-    return(img)
+    return(mask)
        
 def inference(img, lang):
     if lang == "english":