from datasets import load_dataset,concatenate_datasets from setfit import SetFitModel, SetFitTrainer from sentence_transformers.losses import CosineSimilarityLoss