update ckpt and update process

app.py

@@ -24,8 +24,8 @@ import gradio as gr
 # Register refcoco task
 tasks.register_task('ocr', OcrTask)
 
-os.system('wget https://shuangqing-multimodal.oss-cn-zhangjiakou.aliyuncs.com/ocr_general.pt; '
-          'mkdir -p checkpoints; mv ocr_general.pt checkpoints/ocr.pt')
+os.system('wget https://shuangqing-multimodal.oss-cn-zhangjiakou.aliyuncs.com/ocr_general_clean.pt; '
+          'mkdir -p checkpoints; mv ocr_general_clean.pt checkpoints/ocr.pt')
 
 # turn on cuda if GPU is available
 use_cuda = torch.cuda.is_available()
@@ -189,7 +189,7 @@ description = "Gradio Demo for OFA-OCR. Upload your own image or click any one o
               "\"Submit\" and then wait for the generated OCR result.  "
 article = "<p style='text-align: center'><a href='https://github.com/OFA-Sys/OFA' target='_blank'>OFA Github " \
           "Repo</a></p> "
-examples = [['lihe.png'], ['chinese.jpg'], ['paibian.jpeg'], ['shupai.png'], ['zuowen.jpg']]
+examples = [['lihe.png']] #, ['chinese.jpg'], ['paibian.jpeg'], ['shupai.png'], ['zuowen.jpg']]
 io = gr.Interface(fn=ocr, inputs=gr.inputs.Image(type='pil'),
                   outputs=[gr.outputs.Image(type='pil'), gr.outputs.Textbox(label="OCR result")],
                   title=title, description=description, article=article, examples=examples,

ezocr/easyocrlite/reader.py

@@ -104,7 +104,7 @@ class ReaderLite(object):
 
     def process(
         self,
-        image_path: str,
+        img,
         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 = Image.open(image_path).convert('RGB')
+        image = img.convert('RGB')
 
         tensor, inverse_ratio = self.preprocess(
             image, max_size, expand_ratio, sharp, contrast