do it normally

app.py

@@ -49,25 +49,24 @@ 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
 
-# def image_to_base64(image):
-#     buffered = io.BytesIO()
-#     image.save(buffered, format="PNG")
-#     return base64.b64encode(buffered.getvalue()).decode()
-
-
-def numpy_to_pil(image_np):
-    return Image.fromarray(image_np.astype('uint8'), 'RGB')
+def process_input_image(image):
+    if isinstance(image, np.ndarray):
+        return Image.fromarray(image.astype('uint8'), 'RGB')
+    elif isinstance(image, str):
+        return Image.open(image)
+    else:
+        raise ValueError("Unsupported image type")
 
 @spaces.GPU
-def process_image(image_np, task, ocr_type=None, ocr_box=None, ocr_color=None):
+def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None):
     try:
-        if image_np is None:
+        if image is None:
             return "No image provided", None
 
-        image = numpy_to_pil(image_np)
+        pil_image = process_input_image(image)
 
         with io.BytesIO() as buffer:
-            image.save(buffer, format="PNG")
+            pil_image.save(buffer, format="PNG")
             image_path = "/tmp/temp_image.png"
             with open(image_path, "wb") as f:
                 f.write(buffer.getvalue())