thak123 commited on
Commit
80266c4
1 Parent(s): f286b50

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -4,4 +4,5 @@ torch==1.8.1
4
  transformers==4.8.2
5
  pytorch-lightning==1.4.9
6
  tokenizers==0.10.3
7
- numpy==1.21.2
 
 
4
  transformers==4.8.2
5
  pytorch-lightning==1.4.9
6
  tokenizers==0.10.3
7
+ numpy==1.21.2
8
+ scikit-learn==0.24.1