muqtasid87 commited on
Commit
9d3f1be
·
verified ·
1 Parent(s): 383b9f0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  --find-links https://download.pytorch.org/whl/torch_stable.html
2
- torch==2.2.0+cpu
 
3
  Pillow
4
  transformers
5
  timm
@@ -11,6 +12,5 @@ streamlit
11
  numpy
12
  pandas
13
  matplotlib
14
- torchvision
15
  gradio
16
 
 
1
  --find-links https://download.pytorch.org/whl/torch_stable.html
2
+ pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
3
+
4
  Pillow
5
  transformers
6
  timm
 
12
  numpy
13
  pandas
14
  matplotlib
 
15
  gradio
16