Add missing deps

adamjstewart · adamjstewart · commit f0965705084e · 2024-09-02T15:45:18.000+02:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,10 +19,12 @@ classifiers = [
 dependencies = [
     'efficientnet-pytorch==0.7.1',
     'huggingface-hub>=0.24.6',
+    'numpy',
     'pillow',
     'pretrainedmodels==0.7.4',
     'six',
     'timm==0.9.7',
+    'torch',
     'torchvision>=0.5',
     'tqdm',
 ]
