torch
sentence-transformers
argparse
