Update modeling_deepseekocr.py

modeling_deepseekocr.py

@@ -27,7 +27,9 @@ import time
 def load_image(image_path):
 
     try:
-        image = Image.open(image_path)
+        image = image_path
+        if not isinstance(image_path, Image.Image):
+            image = Image.open(image_path)
         
         corrected_image = ImageOps.exif_transpose(image)
         
@@ -722,7 +724,7 @@ class DeepseekOCRForCausalLM(DeepseekV2ForCausalLM):
                     # "content": "<image>\nFree OCR. ",
                     # "content": "<image>\nParse the figure. ",
                     # "content": "<image>\nExtract the text in the image. ",
-                    "images": [image_file] if isinstance(image_file, BytesIO) else [f'{image_file}'],
+                    "images": [image_file] if isinstance(image_file, (BytesIO, Image.Image)) else [f'{image_file}'],
                 },
                 {"role": "<|Assistant|>", "content": ""},
             ]