jeffaudi commited on
Commit
1377fc6
1 Parent(s): 6715ea2

Removing extra pip install in app.py

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +3 -3
app.py CHANGED
@@ -15,7 +15,6 @@ import warnings
15
  import torch
16
 
17
  os.system("python setup.py build develop --user")
18
- os.system("pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118")
19
  os.system("pip install packaging==21.3")
20
  warnings.filterwarnings("ignore")
21
 
 
15
  import torch
16
 
17
  os.system("python setup.py build develop --user")
 
18
  os.system("pip install packaging==21.3")
19
  warnings.filterwarnings("ignore")
20
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- torch
2
- torchvision
3
- transformers
4
  addict
5
  yapf
6
  timm
 
1
+ torch==2.2.1
2
+ torchvision==0.17.1
3
+ transformers==4.38.1
4
  addict
5
  yapf
6
  timm