Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import ImageDraw
 import gradio as gr
 import torch
 import easyocr
-import cv2 as cv
+import cv2
 import math
 import numpy as np
 
@@ -22,6 +22,10 @@ def draw_mask(img, bounds):
         p0, p1, p2, p3 = bound[0]
         draw.polygon((*p0, *p1, *p2, *p3, *p0), fill=255)
     return img
+    
+def masking(img, mask):
+    img = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
+    return img
        
 def inference(img, lang):
     if lang == "english":
@@ -37,22 +41,21 @@ def inference(img, lang):
     im = PIL.Image.open(img.name)
     #mask = PIL.Image.open(img.name)
     mask = Image.new("L", im.size, 0)
-    
     draw_mask(mask, bounds)
+    
+    final = masking(im, mask)
     #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_NS)
     #remove_text(im, mask, bounds)
     lang = ""
     im.save('result.jpg')
     mask.save('mask.png')
+    final.save('final.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)
+    #dst = cv.inpaint(img,mask,3,cv.INPAINT_TELEA)
     #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_TELEA)
     #cv2.imwrite('dst.png', dst)
     #dst.save('dst.png')
-    return (dst)
-    #return ['result.jpg', 'dst.png', pd.DataFrame(bounds). iloc[: , 1:2]]
+    return ['result.jpg', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'
 description = 'Gradio demo for EasyOCR. EasyOCR demo supports 80+ languages.To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to load them. Read more at the links below.'
@@ -68,10 +71,10 @@ choices = [
 gr.Interface(
     inference,
     [gr.inputs.Image(type='file', label='Input'),gr.inputs.Dropdown(choices, type="value", default="korean", label='language')],
-    gr.outputs.Image(type='file', label='Output'), 
-    #gr.outputs.Image(type='file', label='Output'),
+    [gr.outputs.Image(type='file', label='Output'), 
+    gr.outputs.Image(type='file', label='Output'),
     #gr.outputs.Image(type='file', label='Output'), 
-    #gr.outputs.Dataframe()],
+    gr.outputs.Dataframe()],
     title=title,
     description=description,
     article=article,