fix: add torchvision in deps

vra · Jul 8, 2023 · 5e8e43e · 5e8e43e
1 parent 3593d0e
commit 5e8e43e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,10 +18,12 @@ classifiers = [
 ]
 dependencies = [
 "appdirs",
-"torch",
-"pillow",
-"numpy",
 "loguru",
+"numpy",
+"pillow",
+"torch",
+"torchvision",
+"tqdm",
 ]
 
 [project.scripts]