Update app.py

app.py

@@ -25,12 +25,13 @@ def midpoint(x1, y1, x2, y2):
 
 def draw_mask(img, bounds):
     mask = np.zeros(img.shape[:2], dtype="uint8")
-    for box in bounds:
+    for bound in bounds:
+        box0, box1, box2, box3 = bound[0]
         
-        x0, y0 = box[0]
-        x1, y1 = box[1] 
-        x2, y2 = box[2]
-        x3, y3 = box[3] 
+        x0, y0 = box0
+        x1, y1 = box1 
+        x2, y2 = box2
+        x3, y3 = box3 
         
         x_mid0, y_mid0 = midpoint(x1, y1, x2, y2)
         x_mid1, y_mi1 = midpoint(x0, y0, x3, y3)
@@ -59,7 +60,7 @@ def inference(img, lang):
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
     
-    img_array = np.array(im)
+    img_array = asarray(im)
     
     #mask = PIL.Image.open(img.name)
     #mask = Image.new("L", im.size, 0)