Experimenting with few-shot classification.

app.py

@@ -71,7 +71,7 @@ ort_model = ORTModelForFeatureExtraction.from_pretrained('BAAI/bge-small-en-v1.5
 few_shot_model = SetFitModel.from_pretrained("moshew/bge-small-en-v1.5_setfit-sst2-english")
 
 # Train few_shot_model
-candidate_labels = ["supported", "refuted", "neither"]
+candidate_labels = ["supported", "refuted"]
 reference_dataset = load_dataset("emotion")
 dummy_dataset = Dataset.from_dict({})
 train_dataset = get_templated_dataset(dummy_dataset, candidate_labels=candidate_labels, sample_size=8, template="This hypothesis is {}")
@@ -109,8 +109,13 @@ def zero_shot_classification(data):
     response_string = json.dumps(results)
     return response_string
 
+def create_sequences(data):
+    return ['###Given:\n' + data['sequence'] + '\n###Hypothesis:\n' + data['hypothesis_template'].format(label) for label in data['candidate_labels']]
+
 def few_shot_classification(data):
-    results = onnx_few_shot_model(data['sequence'])
+    sequences = create_sequences(data)
+    print(sequences)
+    results = onnx_few_shot_model(sequences)
     probs = onnx_few_shot_model.predict_proba(data['sequence'])
     print(results)
     print(probs)