Update app.py

app.py

@@ -20,6 +20,7 @@ def draw_mask(img, bounds, color='yellow', width=2):
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
         
+        """
         x0,y0 = p0
         x1,y1 = p1
         x2,y2 = p2
@@ -32,7 +33,9 @@ def draw_mask(img, bounds, color='yellow', width=2):
         mask = np.zeros(img.shape[:2], dtype="uint8")
         cv2.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
         img = cv2.inpaint(img, mask, 7, cv2.INPAINT_NS)
-        #draw.polygon((*p0, *p1, *p2, *p3), fill=(200,0,0)) 
+        """
+        
+        draw.polygon((*p0, *p1, *p2, *p3), fill=255) 
     
     #masking
        
@@ -54,6 +57,7 @@ def inference(img, lang):
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
+    mask = Image.new("L", im.size, 0)
     #w, h = im.size 
     
     draw_mask(mask, bounds)