milyiyo commited on
Commit
e5a3796
1 Parent(s): 69eedfd

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +15 -2
requirements.txt CHANGED
@@ -3,7 +3,20 @@ torch
3
  diffusers==0.1.1
4
  opencv-python
5
 
 
6
  transformers
7
- pillow
 
 
 
 
 
 
 
8
  timm
9
- numpy
 
 
 
 
 
 
3
  diffusers==0.1.1
4
  opencv-python
5
 
6
+ # For captioning
7
  transformers
8
+ h5py
9
+ pandas
10
+ tqdm
11
+ scikit-image
12
+ ipywidgets
13
+ wandb
14
+ bert-score
15
+ ftfy
16
  timm
17
+ lmdbdict
18
+ yacs
19
+ pyemd
20
+ gensim
21
+ pytorch-lightning==1.0.0
22
+ gdown