Update app.py

app.py

@@ -43,10 +43,6 @@ def draw_mask(img, bounds):
     img = cv.inpaint(img, mask, 3, cv.INPAINT_TELEA)
                  
     return(img)
-    
-def masking(img, mask):
-    #img = cv.inpaint(img, mask, 3, cv.INPAINT_TELEA)
-    return img
        
 def inference(img, lang):
     if lang == "english":
@@ -60,38 +56,18 @@ def inference(img, lang):
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
-    
     img_array = np.array(im)
-    
-    #mask = PIL.Image.open(img.name)
-    #mask = Image.new("L", im.size, 0)
     im = draw_mask(img_array, bounds)
     im = Image.fromarray(im, 'RGB')
-    #
-    #masking(img_array, mask)
-    #img_inpainted = cv.inpaint(img_array, mask, 7, cv.INPAINT_NS)
-    #remove_text(im, mask, bounds)
     lang = ""
-    #cv.imshow('dst', dst)
     im.save('result.jpg')
-    #mask.save('mask.png')
-    
-    #img = cv.imread('result.jpg')
-    #mask = cv.imread('mask2.png',0)
-    #dst = cv.inpaint(img,mask,3,cv.INPAINT_TELEA)
     
-    #final.save('final.png')
-    #img_inpainted.save('inpaint.jpg')
-    #dst = cv2.inpaint(np.float32(img),np.float32(mask),3,cv2.INPAINT_TELEA)
-    #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_TELEA)
-    #cv2.imwrite('dst.png', dst)
-    #cv.imwrite('result.jpg',dst)
     return ['result.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.'
 article = "<p style='text-align: center'><a href='https://www.jaided.ai/easyocr/'>Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.</a> | <a href='https://github.com/JaidedAI/EasyOCR'>Github Repo</a></p>"
-css = ".output_image, .input_image {height: 300% !important; width: 40rem !important;}"
+css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
 examples = [['korean.jpg',"korean"]]
 choices = [
     "chinese",
@@ -101,7 +77,7 @@ choices = [
 ]
 gr.Interface(
     inference,
-    [gr.inputs.Image(type='file', shape = (1700,2400), label='Input'),gr.inputs.Dropdown(choices, type="value", default="korean", label='language')],
+    [gr.inputs.Image(type='pil', 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'),