add back loadimg

app.py

@@ -47,18 +47,19 @@ 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=False):
+    img = load_img(image, output_type="pil", input_type="auto")
     if task == "Plain Text OCR":
-        res = model.chat(tokenizer, image, ocr_type='ocr')
+        res = model.chat(tokenizer, img, ocr_type='ocr')
     elif task == "Format Text OCR":
-        res = model.chat(tokenizer, image, ocr_type='format')
+        res = model.chat(tokenizer, img, ocr_type='format')
     elif task == "Fine-grained OCR (Box)":
-        res = model.chat(tokenizer, image, ocr_type=ocr_type, ocr_box=ocr_box)
+        res = model.chat(tokenizer, img, ocr_type=ocr_type, ocr_box=ocr_box)
     elif task == "Fine-grained OCR (Color)":
-        res = model.chat(tokenizer, image, ocr_type=ocr_type, ocr_color=ocr_color)
+        res = model.chat(tokenizer, img, ocr_type=ocr_type, ocr_color=ocr_color)
     elif task == "Multi-crop OCR":
-        res = model.chat_crop(tokenizer, image_file=image)
+        res = model.chat_crop(tokenizer, image_file=img)
     elif task == "Render Formatted OCR":
-        res = model.chat(tokenizer, image, ocr_type='format', render=True, save_render_file='./results/demo.html')
+        res = model.chat(tokenizer, img, 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
@@ -157,4 +158,5 @@ with gr.Blocks() as demo:
         outputs=[output_text, output_html]
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()