Update app.py

app.py

@@ -54,12 +54,10 @@ def inference(img, lang):
     im = PIL.Image.open(img.name)
     img_array = np.array(im)
     img = draw_mask(img_array, bounds)
-    img = Image.fromarray(im, 'RGB')
+    img = Image.fromarray(img, 'RGB')
     lang = ""
-    im.save('box.jpb')
     img.save('result.jpg')
-    
-    return ['box.jpg', 'result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
+    return ['result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'Manga Image Cleaner'
 description = 'Image inpainting and text detection demo with the use of EasyOCR and CV2. To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to test the program.'
@@ -77,7 +75,7 @@ gr.Interface(
     [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.Dataframe(header=["TEXT"])],
     title=title,
     description=description,
     article=article,