jpterry commited on
Commit
e157aa1
·
1 Parent(s): 8f30ca3

timm to make models

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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