Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import torch
 import easyocr
 import omegaconf
-import numpy
+import cv2
 
 from vietocr.vietocr.tool.predictor import Predictor
 from vietocr.vietocr.tool.config import Cfg
@@ -30,7 +30,7 @@ def draw_boxes(image, bounds, color='yellow', width=2):
     return image
 
 def inference(filepath, lang):
-    img = Image.open(filepath).convert("RGB")
+    img = cv2.imread(filepath)
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(filepath)
     new_bounds=[]
@@ -47,9 +47,8 @@ def inference(filepath, lang):
         max_y = max(tl[1], tr[1], br[1], bl[1])
         # crop the region of interest (ROI)
         
-        cropped_image = img.crop((min_x, min_y, max_x, max_y)) # crop the image
+        cropped_image = img[(min_x:(max_x - min_x), min_y:(max_y - min_y)] # crop the image
         cropped_image = Image.fromarray(cropped_image)
-        cropped_image = cropped_image.convert("RGB")
         out = recognitor.predict(cropped_image)
         print(out)
         new_bounds.append((bbox,text, out, prob))