timm to make models
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -9,7 +9,7 @@ from scipy import special
|
|
9 |
import sys
|
10 |
import timm
|
11 |
from types import SimpleNamespace
|
12 |
-
from transformers import AutoModel, pipeline
|
13 |
import torch
|
14 |
|
15 |
sys.path.insert(1, "../")
|
|
|
9 |
import sys
|
10 |
import timm
|
11 |
from types import SimpleNamespace
|
12 |
+
# from transformers import AutoModel, pipeline
|
13 |
import torch
|
14 |
|
15 |
sys.path.insert(1, "../")
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ matplotlib
|
|
4 |
scipy
|
5 |
Pillow
|
6 |
transformers
|
7 |
-
torchvision
|
|
|
|
4 |
scipy
|
5 |
Pillow
|
6 |
transformers
|
7 |
+
torchvision
|
8 |
+
timm
|