add loadimg !!!!!!

app.py

@@ -4,6 +4,7 @@ from transformers import AutoModel, AutoTokenizer, AutoConfig
 import os
 import base64
 import spaces
+from loadimg import load_img
 
 title = """# 🙋🏻‍♂️Welcome to Tonic's🫴🏻📸GOT-OCR"""
 description = """"
@@ -44,9 +45,10 @@ model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True,
 model = model.eval().cuda()
 model.config.pad_token_id = tokenizer.eos_token_id
 
-
 @spaces.GPU
-def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=True):
+def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=False):
+    # Load the image using loadimg
+    img = load_img(image, output_type="pil", input_type="auto")
     if task == "Plain Text OCR":
         res = model.chat(tokenizer, image, ocr_type='ocr')
     elif task == "Format Text OCR":
@@ -58,8 +60,8 @@ def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, rend
     elif task == "Multi-crop OCR":
         res = model.chat_crop(tokenizer, image_file=image)
     elif task == "Render Formatted OCR":
-        res = model.chat(tokenizer, image, ocr_type='format', render=True, save_render_file='./demo.html')
-        with open('./demo.html', 'r') as f:
+        res = model.chat(tokenizer, image, ocr_type='format', render=True, save_render_file='./results/demo.html')
+        with open('./results/demo.html', 'r') as f:
             html_content = f.read()
         return res, html_content
     
@@ -85,8 +87,8 @@ def update_inputs(task):
     elif task == "Render Formatted OCR":
         return [gr.update(visible=False)] * 3 + [gr.update(visible=True)]
     
-def ocr_demo(image, task, ocr_type, ocr_box, ocr_color, render):
-    res, html_content = process_image(image, task, ocr_type, ocr_box, ocr_color, render)
+def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
+    res, html_content = process_image(image, task, ocr_type, ocr_box, ocr_color)
     if html_content:
         return res, html_content
     return res, None