From 5fe7858a556a11cde96bfd61b9b187080c82505d Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Sun, 9 Jan 2022 15:54:30 -0300 Subject: [PATCH] fix project layout --- rembg/detect.py | 2 +- setup.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) 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")