Add requirement
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -2,4 +2,5 @@ torch
|
|
| 2 |
torchvision
|
| 3 |
Pillow
|
| 4 |
numpy
|
| 5 |
-
|
|
|
|
|
|
| 2 |
torchvision
|
| 3 |
Pillow
|
| 4 |
numpy
|
| 5 |
+
timm
|
| 6 |
+
pyyaml
|