kitkeat's picture
Update requirements.txt
c0bdfaf
#torch
#transformers
transformers==4.18.0
# numpy==1.21.6
flask==2.2.2
torch==1.12.1
pytorch-lightning==1.6.5
scikit-learn==1.0.2
datasets==2.1.0
pandas==1.3.5