@@ -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):
@@ -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")