Update app.py

app.py

@@ -9,7 +9,9 @@ import easyocr
 import omegaconf
 import cv2
 import json
+from autocorrect import Speller
 
+spell = Speller(lang='vi')
 from vietocr.vietocr.tool.predictor import Predictor
 from vietocr.vietocr.tool.config import Cfg
 
@@ -86,7 +88,7 @@ def inference(filepath, lang, labels):
         cropped_image = img[min_y:max_y,min_x:max_x] # crop the image
         cropped_image = Image.fromarray(cropped_image)
         out = recognitor.predict(cropped_image)
-        texts = texts + '\t' + out
+        texts = texts + '\t' + spell(out)
     result = query(texts, labels)
     jsonText = json.dumps(result)
     im = PIL.Image.open(filepath)