Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Removing extra pip install in app.py
Browse files- app.py +0 -1
- 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
|