DerrylNessie commited on
Commit
1b1fded
1 Parent(s): 6ba9578

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -35,15 +35,15 @@ def inference(img, lang):
35
  reader = easyocr.Reader(lang)
36
  bounds = reader.readtext(img.name)
37
  im = PIL.Image.open(img.name)
38
- mask = PIL.Image.open(img.name)
39
  #mask = Image.new("L", im.size, 0)
40
 
41
- #draw_mask(im, bounds)
42
  #remove_text(im, mask, bounds)
43
  lang = ""
44
  im.save('result.jpg')
45
- mask.save('mask.png')
46
- return ['result.jpg', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
47
 
48
  title = 'EasyOCR'
49
  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 +59,7 @@ choices = [
59
  gr.Interface(
60
  inference,
61
  [gr.inputs.Image(type='file', label='Input'),gr.inputs.Dropdown(choices, type="value", default="korean", label='language')],
62
- [gr.outputs.Image(type='file', label='Output'), gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe()],
63
  title=title,
64
  description=description,
65
  article=article,
 
35
  reader = easyocr.Reader(lang)
36
  bounds = reader.readtext(img.name)
37
  im = PIL.Image.open(img.name)
38
+ #mask = PIL.Image.open(img.name)
39
  #mask = Image.new("L", im.size, 0)
40
 
41
+ draw_mask(im, bounds)
42
  #remove_text(im, mask, bounds)
43
  lang = ""
44
  im.save('result.jpg')
45
+ #mask.save('mask.png')
46
+ return ['result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
47
 
48
  title = 'EasyOCR'
49
  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
  gr.Interface(
60
  inference,
61
  [gr.inputs.Image(type='file', label='Input'),gr.inputs.Dropdown(choices, type="value", default="korean", label='language')],
62
+ [gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe()],
63
  title=title,
64
  description=description,
65
  article=article,