Explorar o código

Merge pull request #48 from Doge2077/master

修复非标准图片base64格式导致图像显示错误
Qing Yan hai 3 meses
pai
achega
e032665f4c
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      dots_ocr/utils/image_utils.py

+ 1 - 1
dots_ocr/utils/image_utils.py

@@ -68,7 +68,7 @@ def PILimage_to_base64(image, format='PNG'):
     buffered = BytesIO()
     image.save(buffered, format=format)
     base64_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
-    return f"data:image;base64,{base64_str}"
+    return f"data:image/{format.lower()};base64,{base64_str}"
 
 
 def to_rgb(pil_image: Image.Image) -> Image.Image: