datasets
sentence-transformers
torch
pandas