Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +15 -2
requirements.txt
CHANGED
@@ -3,7 +3,20 @@ torch
|
|
3 |
diffusers==0.1.1
|
4 |
opencv-python
|
5 |
|
|
|
6 |
transformers
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
timm
|
9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
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
|