Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import gradio as gr
 import omegaconf
 import torch
+import cv2
+
 import easyocr
 from PIL import Image
 
@@ -32,8 +34,9 @@ def viet_ocr_predict(inp):
     return out
 def predict(filepath):
     bounds = reader.readtext(filepath)
-    inp = Image.open(filepath).convert('RGB')
-    width, height = inp.size
+    inp = cv2.imread(filepath)
+
+    width, height, _ = inp.shape
     texts=''
     for (bbox, text, prob) in bounds:
         (tl, tr, br, bl) = bbox
@@ -51,9 +54,6 @@ def predict(filepath):
         max_y = max(tl[1], tr[1], br[1], bl[1])
         max_y = min(height-1, max_y)
         # crop the region of interest (ROI)
-
-
-
         cropped_image = inp[min_y:max_y,min_x:max_x] # crop the image
         cropped_image = Image.fromarray(cropped_image)
         out = viet_ocr_predict(cropped_image)