Image Classification
Transformers
English
art
benjaminStreltzin commited on
Commit
f6991e3
·
verified ·
1 Parent(s): 134bd4d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
- torch
2
- torchvision
3
- transformers
 
4
  Pillow
5
  pandas
6
  scikit-learn
 
1
+ # Requires Python 3.10.14
2
+ torch==2.3.0+cu121
3
+ torchvision==0.18.0
4
+ transformers==4.40.2
5
  Pillow
6
  pandas
7
  scikit-learn