Update app.py

app.py

@@ -50,12 +50,12 @@ def inference(filepath, lang):
         # crop the region of interest (ROI)
         img = Image.open(filepath)
         #img = img[y0:y1, x0:x1]
-        img = img.crop((x1, y1, x3, y3)) # crop the image
+        img = img.crop((max(0,x1-5), max(y1-5), min(x3+5,width), min(y3+5, height)) # crop the image
         img = process_input(img, config['dataset']['image_height'], 
                     config['dataset']['image_min_width'], config['dataset']['image_max_width'])
         out = translate(img, model)[0].tolist()
         out = vocab.decode(out)
-        new_bounds.append((bbox, out, prob))
+        new_bounds.append((bbox,text, out, prob))
     im = PIL.Image.open(filepath)
     draw_boxes(im, bounds)
     im.save('result.jpg')
@@ -67,94 +67,12 @@ article = "<p style='text-align: center'><a href='https://www.jaided.ai/easyocr/
 examples = [['english.png',['en']],['thai.jpg',['th']],['french.jpg',['fr', 'en']],['chinese.jpg',['ch_sim', 'en']],['japanese.jpg',['ja', 'en']],['korean.png',['ko', 'en']],['Hindi.jpeg',['hi', 'en']]]
 css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
 choices = [
-    "abq",
-    "ady",
-    "af",
-    "ang",
-    "ar",
-    "as",
-    "ava",
-    "az",
-    "be",
-    "bg",
-    "bh",
-    "bho",
-    "bn",
-    "bs",
-    "ch_sim",
-    "ch_tra",
-    "che",
-    "cs",
-    "cy",
-    "da",
-    "dar",
-    "de",
-    "en",
-    "es",
-    "et",
-    "fa",
-    "fr",
-    "ga",
-    "gom",
-    "hi",
-    "hr",
-    "hu",
-    "id",
-    "inh",
-    "is",
-    "it",
-    "ja",
-    "kbd",
-    "kn",
-    "ko",
-    "ku",
-    "la",
-    "lbe",
-    "lez",
-    "lt",
-    "lv",
-    "mah",
-    "mai",
-    "mi",
-    "mn",
-    "mr",
-    "ms",
-    "mt",
-    "ne",
-    "new",
-    "nl",
-    "no",
-    "oc",
-    "pi",
-    "pl",
-    "pt",
-    "ro",
-    "ru",
-    "rs_cyrillic",
-    "rs_latin",
-    "sck",
-    "sk",
-    "sl",
-    "sq",
-    "sv",
-    "sw",
-    "ta",
-    "tab",
-    "te",
-    "th",
-    "tjk",
-    "tl",
-    "tr",
-    "ug",
-    "uk",
-    "ur",
-    "uz",
     "vi"
 ]
 gr.Interface(
     inference,
-    [gr.inputs.Image(type='filepath', label='Input'),gr.inputs.CheckboxGroup(choices, type="value", default=['en'], label='language')],
-    [gr.outputs.Image(type='pil', label='Output'), gr.outputs.Dataframe(type='pandas', headers=['text', 'confidence'])],
+    [gr.inputs.Image(type='filepath', label='Input'),gr.inputs.CheckboxGroup(choices, type="value", default=['vi'], label='language')],
+    [gr.outputs.Image(type='pil', label='Output'), gr.outputs.Dataframe(type='pandas', headers=['easyOCR','vietOCR', 'confidence'])],
     title=title,
     description=description,
     article=article,