diff --git a/rembg/detect.py b/rembg/detect.py index 8cc744b..b6fbcac 100644 --- a/rembg/detect.py +++ b/rembg/detect.py @@ -16,7 +16,7 @@ from torchvision import transforms from tqdm import tqdm from .data_loader import RescaleT, ToTensorLab -from .u2net import U2NETP, U2NET +from .u2net import U2NET, U2NETP def download_file_from_google_drive(id, fname, destination): diff --git a/setup.py b/setup.py index aebdf50..d4fb7a4 100644 --- a/setup.py +++ b/setup.py @@ -1,12 +1,12 @@ +import os import pathlib import sys -import os sys.path.append(os.path.dirname(__file__)) -import versioneer - from setuptools import find_packages, setup +import versioneer + here = pathlib.Path(__file__).parent.resolve() long_description = (here / "README.md").read_text(encoding="utf-8")