valegro commited on
Commit
d06be5a
·
verified ·
1 Parent(s): 3962476

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -2
requirements.txt CHANGED
@@ -1,3 +1,12 @@
1
- altair
 
 
 
 
 
 
2
  pandas
3
- streamlit
 
 
 
 
1
+ torch>=2.3 # include CUDA in build GPU
2
+ transformers>=4.41
3
+ segment-anything==1.0.2
4
+ timm
5
+ opencv-python-headless
6
+ pillow
7
+ numpy
8
  pandas
9
+ scikit-learn
10
+ seaborn
11
+ streamlit
12
+ huggingface-hub