2
0
Daniel Gatis 3 жил өмнө
parent
commit
5fe7858a55
2 өөрчлөгдсөн 4 нэмэгдсэн , 4 устгасан
  1. 1 1
      rembg/detect.py
  2. 3 3
      setup.py

+ 1 - 1
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):

+ 3 - 3
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")