From 5e8e43e112826cd4e41dbbca3482ab812036e113 Mon Sep 17 00:00:00 2001
From: vra <wyf.brz@gmail.com>
Date: Sat, 8 Jul 2023 14:42:00 +0800
Subject: [PATCH] fix: add torchvision in deps

---
 pyproject.toml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 70c5a92..449d9a7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,10 +18,12 @@ classifiers = [
 ]
 dependencies = [
 "appdirs",
-"torch",
-"pillow",
-"numpy",
 "loguru",
+"numpy",
+"pillow",
+"torch",
+"torchvision",
+"tqdm",
 ]
 
 [project.scripts]