requirements.txt

flask == 2.2.2
pytest
torch == 2.0.1
transformers == 4.31.0

# for the tokenizer of the baseline model
protobuf == 4.24.0
sentencepiece==0.1.99