add torchvision requirements

deplay · deplay · commit d9864ad6e7ed · 2022-06-30T17:33:41.000+08:00
diff --git a/requirements.txt b/requirements.txt
@@ -17,4 +17,5 @@ rouge
 tqdm>=4.48.2
 pillow
 albumentations
-einops
+einops
+torchvision
