fix: folder hash error (#19)

Final2x · Jun 28, 2024 · ce4ca12 · ce4ca12
1 parent 2fa106f
commit ce4ca12
Show file tree

Hide file tree

Showing 4 changed files with 58 additions and 119 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,13 +18,15 @@ dependencies = [
   "realesrgan-ncnn-py>=2.0.0",
   "waifu2x-ncnn-py>=2.0.0",
   "PyYAML~=6.0",
-  "loguru~=0.7.0"
+  "loguru~=0.7.0",
+  "requests~=2.26.0",
+  "tqdm~=4.62.0"
 ]
 license = {text = "BSD-3-Clause"}
 name = "Final2x-core"
 readme = "README.md"
 requires-python = ">=3.8, <3.13"
-version = "2.0.1"
+version = "2.0.2"
 
 [project.scripts]
 Final2x-core = "Final2x_core.__main__:main"
@@ -53,8 +55,6 @@ test = [
   "pytest",
   "pytest-cov",
   "scikit-image",
-  "requests",
-  "tqdm",
   "types-PyYAML",
   "types-requests",
   "pyinstaller~=6.3"

diff --git a/src/Final2x_core/__main__.py b/src/Final2x_core/__main__.py
@@ -5,7 +5,7 @@
 
 from loguru import logger
 
-from Final2x_core.download_models import download_all, recheck_all
+from Final2x_core.download_models import download_all
 from Final2x_core.src.SRqueue import SR_queue
 from Final2x_core.src.utils.getConfig import SRCONFIG
 
@@ -50,7 +50,9 @@ def open_folder(path: str) -> None:
 
 
 def main() -> None:
-    recheck_all()  # check if models are downloaded
+    entries = os.listdir(projectPATH / "models")
+    if len(entries) <= 1:  # first time run
+        download_all()
 
     if args.LOG:
         # init logger