Update app.py

app.py

@@ -11,6 +11,13 @@ import numpy as np
 
 torch.hub.download_url_to_file('https://i.pinimg.com/originals/45/d0/30/45d03054e15f4be731781eecba7458a4.jpg', 'korean.jpg')
 
+def convert(img):
+    pil_im = Image.fromarray(image)
+    b = io.BytesIO()
+    pil_im.save(b, 'jpeg')
+    im_bytes = b.getvalue()
+    return im_bytes
+    
 def midpoint(x1, y1, x2, y2):
     x_mid = int((x1 + x2)/2)
     y_mid = int((y1 + y2)/2)
@@ -39,6 +46,10 @@ def inference(img, lang):
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
+    
+    byte_img = convert(im)
+    img_array = np.array(Image.open(io.BytesIO(byte_img)))
+    
     #mask = PIL.Image.open(img.name)
     mask = Image.new("L", im.size, 0)
     draw_mask(mask, bounds)
@@ -51,17 +62,16 @@ def inference(img, lang):
     im.save('result.jpg')
     mask.save('mask.png')
     
-    img = cv.imread('result.jpg')
-    mask = cv.imread('mask2.png',0)
-    dst = cv.inpaint(img,mask,3,cv.INPAINT_TELEA)
-    
+    #img = cv.imread('result.jpg')
+    #mask = cv.imread('mask2.png',0)
+    #dst = cv.inpaint(img,mask,3,cv.INPAINT_TELEA)
     
     #final.save('final.png')
     #img_inpainted.save('inpaint.jpg')
     #dst = cv2.inpaint(np.float32(img),np.float32(mask),3,cv2.INPAINT_TELEA)
     #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_TELEA)
     #cv2.imwrite('dst.png', dst)
-    dst.save('result.jpg')
+    #cv.imwrite('result.jpg',dst)
     return ['result.jpg', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'