Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import ImageDraw
 import gradio as gr
 import torch
 import easyocr
-import cv2
+import cv2 as cv
 import math
 import numpy as np
 
@@ -42,7 +42,7 @@ def inference(img, lang):
     #mask = PIL.Image.open(img.name)
     mask = Image.new("L", im.size, 0)
     draw_mask(mask, bounds)
-    dst = cv2.inpaint(np.float32(img),np.float32(mask),3,cv2.INPAINT_TELEA)
+    #
     #masking(im, mask)
     #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_NS)
     #remove_text(im, mask, bounds)
@@ -50,12 +50,18 @@ def inference(img, lang):
     cv2.imshow('dst', dst)
     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)
+    
+    
     #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('dst.png')
+    dst.save('result.jpg')
     return ['result.jpg', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'