Experimenting with few-shot classification.

app.py

@@ -7,7 +7,7 @@ from optimum.onnxruntime import ORTModelForSequenceClassification
 from optimum.onnxruntime import ORTModelForFeatureExtraction
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from setfit import SetFitModel, Trainer, TrainingArguments
+from setfit import SetFitModel, SetFitTrainer, Trainer, TrainingArguments
 from setfit.exporters.utils import mean_pooling
 from setfit import get_templated_dataset
 from datasets import load_dataset