Update app.py

app.py

@@ -27,26 +27,6 @@ def draw_boxes(img, bounds, color='yellow', width=2):
         cv.fillPoly(mask, pts, color =(255,255,255))
     img = cv.inpaint(img, mask, 3, cv.INPAINT_NS)       
     return(img)
-    
-def draw_mask(img, bounds):
-    mask = np.zeros(img.shape[:2], dtype="uint8")
-    for bound in bounds:
-        box0, box1, box2, box3 = bound[0]
-        
-        x0, y0 = box0
-        x1, y1 = box1 
-        x2, y2 = box2
-        x3, y3 = box3 
-        
-        x_mid0, y_mid0 = midpoint(x1, y1, x2, y2)
-        x_mid1, y_mi1 = midpoint(x0, y0, x3, y3)
-        
-        thickness = int(math.sqrt((x2 - x1)**2 + (y2 - y1)**2))
-        
-        cv.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
-    img = cv.inpaint(img, mask, 1, cv.INPAINT_NS)
-                 
-    return(img)
        
 def inference(img, lang):
     if lang == "english":
@@ -62,14 +42,11 @@ def inference(img, lang):
     im = PIL.Image.open(img.name)
     img_array = np.array(im)
     img = draw_boxes(img_array, bounds)
-    im = draw_mask(img_array, bounds)
     img = Image.fromarray(img, 'RGB')
-    im = Image.fromarray(im, 'RGB')
     lang = ""
     img.save('box.jpg')
-    im.save('result.jpg')
     
-    return ['box.jpg', 'result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
+    return ['box.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.'
@@ -85,8 +62,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.Image(type='file', label='Output'),  
     gr.outputs.Dataframe()],
     title=title,
     description=description,