debug

app.py

@@ -66,7 +66,7 @@ def four_point_transform(image: np.ndarray, rect: FourPoint) -> np.ndarray:
     return warped
 
 
-def get_images(img, reader: ReaderLite, **kwargs):
+def get_images(img: str, reader: ReaderLite, **kwargs):
     results = reader.process(img, **kwargs)
     return results
 
@@ -164,7 +164,7 @@ def apply_half(t):
 
 
 def ocr(img):
-    out_img = img
+    out_img = Image.open(img)
     results = get_images(img, reader)
     box_list, image_list = zip(*results)
     draw_boxes(out_img, box_list)
@@ -191,7 +191,7 @@ description = "Gradio Demo for OFA-OCR. Upload your own image or click any one o
 article = "<p style='text-align: center'><a href='https://github.com/OFA-Sys/OFA' target='_blank'>OFA Github " \
           "Repo</a></p> "
 examples = [['lihe.png']]
-io = gr.Interface(fn=ocr, inputs=gr.inputs.Image(type='pil'),
+io = gr.Interface(fn=ocr, inputs=gr.inputs.Image(type='filepath'),
                   outputs=[gr.outputs.Image(type='pil'), gr.outputs.Textbox(label="OCR result")],
                   title=title, description=description, article=article, examples=examples,
                   allow_flagging=False, allow_screenshot=False)

ezocr/easyocrlite/reader.py

@@ -104,7 +104,7 @@ class ReaderLite(object):
 
     def process(
         self,
-        img,
+        image_path: str,
         max_size: int = 960,
         expand_ratio: float = 1.0,
         sharp: float = 1.0,
@@ -121,7 +121,7 @@ class ReaderLite(object):
         min_size: float = 0.01,
     ) -> Tuple[BoxTuple, list[np.ndarray]]:
 
-        image = img.convert('RGB')
+        image = Image.open(image_path).convert('RGB')
         tensor, inverse_ratio = self.preprocess(
             image, max_size, expand_ratio, sharp, contrast
         )