Update app.py

app.py

@@ -18,21 +18,6 @@ def midpoint(x1, y1, x2, y2):
     y_mid = int((y1 + y2)/2)
     return (x_mid, y_mid)
     
-def draw_boxes(image, bounds):
-    draw = ImageDraw.Draw(image)
-    #font = ImageFont.truetype("sans-serif.ttf", 12)
-    for bound in bounds:
-        boxes = [];
-        count = 1;
-        text = str(count)
-        p0, p1, p2, p3 = bound[0]
-        
-        draw.polygon((*p0, *p1, *p2, *p3), fill="red")
-        draw.text((*p0),"HELLO")
-        
-    bounds.append(boxes) 
-    return image
-    
 def draw_mask(img, bounds):
     mask = np.zeros(img.shape[:2], dtype="uint8")
     for bound in bounds:
@@ -67,14 +52,13 @@ def inference(img, lang):
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
     img_array = np.array(im)
-    draw_boxes(im, bounds)
     img = draw_mask(img_array, bounds)
     img = Image.fromarray(im, 'RGB')
     lang = ""
     im.save('box.jpb')
     img.save('result.jpg')
     
-    return ['box.jpg', 'result.jpg', pd.DataFrame(bounds). iloc[: , [1, 3]]]
+    return ['box.jpg', 'result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'
 description = 'Gradio demo for EasyOCR. EasyOCR demo supports 80+ languages.To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to load them. Read more at the links below.'