torch
sentence-transformers
huggingface-hub
dataset