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
 
@@ -39,12 +39,17 @@ def inference(img, lang):
     mask = Image.new("L", im.size, 0)
     
     draw_mask(mask, bounds)
-    #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_NS)
+    img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_NS)
     #remove_text(im, mask, bounds)
     lang = ""
     im.save('result.jpg')
     mask.save('mask.png')
     #img_inpainted.save('inpaint.jpg')
+    img = cv.imread('result.jpg')
+    mask = cv.imread('mask.png',0)
+    dst = cv.inpaint(img,mask,3,cv.INPAINT_TELEA)
+    #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_TELEA)
+    #dst.save('dst.jpg')
     return ['result.jpg', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'