lfolle commited on
Commit
0b31fb7
1 Parent(s): 1c146a3

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -4,6 +4,8 @@ gradio
4
  scikit-learn
5
  seaborn
6
  huggingface_hub
 
 
7
  pingouin
8
  pyheif
9
  --extra-index-url https://download.pytorch.org/whl/cu116
 
4
  scikit-learn
5
  seaborn
6
  huggingface_hub
7
+ pytorch-lightning==1.6.4
8
+ timm==0.6.7
9
  pingouin
10
  pyheif
11
  --extra-index-url https://download.pytorch.org/whl/cu116