AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-20 02:29:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix linters

Browse Source
This commit is contained in:
Daniel Gatis 2024-05-23 19:25:52 -03:00
parent 747ef5e793
commit bfafe9e9b4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rembg/bg.py
View File

@ -1,6 +1,6 @@
import io import io
from enum import Enum from enum import Enum
from typing import Any, List, Optional, Tuple, Union from typing import Any, List, Optional, Tuple, Union, cast
import numpy as np import numpy as np
import onnxruntime as ort import onnxruntime as ort
@ -192,7 +192,7 @@ def fix_image_orientation(img: PILImage) -> PILImage:
Returns: Returns:
PILImage: The fixed image. PILImage: The fixed image.
""" """
return ImageOps.exif_transpose(img) return cast(PILImage, ImageOps.exif_transpose(img))
def download_models() -> None: def download_models() -> None: