Experimenting with few-shot classification.

app.py

@@ -10,7 +10,7 @@ from fastapi.middleware.cors import CORSMiddleware
 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
+from datasets import load_dataset, Dataset
 
 # CORS Config
 app = FastAPI()