Experimenting with few-shot classification.

app.py

@@ -67,15 +67,16 @@ def get_templated_dataset(candidate_labels, sample_size, template="This sentence
     return examples
 
 candidate_labels = ["true", "false"]
-synthetic_dataset = get_templated_dataset(candidate_labels, sample_size=8)
+synthetic_dataset = get_templated_dataset(candidate_labels=candidate_labels, sample_size=8, template="This statement is {}")
 
 args = TrainingArguments(
     batch_size=32,
     num_epochs=1
 )
 
-trainer = SetFitTrainer(
+trainer = Trainer(
     model=few_shot_model,
+    args=args,
     train_dataset=synthetic_dataset
 )
 trainer.train()