add back loadimg

app.py

@@ -5,6 +5,7 @@ import os
 import base64
 import spaces
 from loadimg import load_img
+from PIL import Image
 
 title = """# 🙋🏻‍♂️Welcome to Tonic's🫴🏻📸GOT-OCR"""
 description = """"
@@ -47,7 +48,12 @@ 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 isinstance(image, str):
+        img_path = image
+    else:
+        img = Image.open(image)
+        img_path = "/tmp/temp_image.png"
+        img.save(img_path)
     if task == "Plain Text OCR":
         res = model.chat(tokenizer, img, ocr_type='ocr')
     elif task == "Format Text OCR":
@@ -64,6 +70,10 @@ def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, rend
             html_content = f.read()
         return res, html_content
     
+    # Clean up 
+    if img_path.startswith("/tmp/"):
+        os.remove(img_path)
+    
     return res, None
 
 def update_inputs(task):
@@ -94,7 +104,6 @@ def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
 
 with gr.Blocks() as demo:
     gr.Markdown(title)
-    # gr.Markdown()
     gr.Markdown(description)
     with gr.Row():
         with gr.Column():