Update app.py

app.py

@@ -39,11 +39,13 @@ 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)
     #remove_text(im, mask, bounds)
     lang = ""
     im.save('result.jpg')
     mask.save('mask.jpg')
-    return ['result.jpg', 'mask.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
+    img_inpainted.save('inpaint.jpg')
+    return ['result.jpg', 'mask.jpg', 'inpaint.jpg', 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.'
@@ -59,7 +61,7 @@ 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.Dataframe()],
+    [gr.outputs.Image(type='file', label='Output'), gr.outputs.Image(type='file', label='Output'), gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe()],
     title=title,
     description=description,
     article=article,