Update modeling_deepseekocr.py

modeling_deepseekocr.py

@@ -27,11 +27,6 @@ import time
 def load_image(image_path):
 
     try:
-        #dummy_img = Image.new("RGB", (640, 640), color="white")
-        #ioBuff = BytesIO()
-        #dummy_img.save(ioBuff, format="PNG")
-        #ioBuff.seek(0)
-        #image = Image.open(ioBuff)
         print(f"image_path:",type(image_path))
         image = Image.open(image_path)
         
@@ -723,7 +718,7 @@ class DeepseekOCRForCausalLM(DeepseekV2ForCausalLM):
                     # "content": "<image>\nFree OCR. ",
                     # "content": "<image>\nParse the figure. ",
                     # "content": "<image>\nExtract the text in the image. ",
-                    "images": [f'{image_file}'],
+                    "images": [image_file] if isinstance(image_file, BytesIO) else [f'{image_file}'],
                 },
                 {"role": "<|Assistant|>", "content": ""},
             ]