Update app.py

app.py

@@ -40,7 +40,7 @@ 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, 7, cv.INPAINT_NS)
+    img = cv.inpaint(img, mask, 7, cv.INPAINT_NS)
                  
     return(img)
     
@@ -66,6 +66,7 @@ def inference(img, lang):
     #mask = PIL.Image.open(img.name)
     #mask = Image.new("L", im.size, 0)
     im = draw_mask(img_array, bounds)
+    im = Image.fromarray(im, 'RGB')
     #
     #masking(img_array, mask)
     #img_inpainted = cv.inpaint(img_array, mask, 7, cv.INPAINT_NS)