Experimenting with few-shot classification.

app.py

@@ -120,7 +120,13 @@ def few_shot_classification(data):
     # results = onnx_few_shot_model(sequences)
     probs = onnx_few_shot_model.predict_proba(sequences)
     scores = [true[0] for true in probs]
-    response_dict = {'scores': scores, 'labels': data['candidate_labels']}
+
+    composite = list(zip(scores, data['candidate_labels']))
+    composite = sorted(composite, key=lambda x: x[0], reverse=True)
+
+    labels, scores = zip(*composite)
+
+    response_dict = {'scores': scores, 'labels': labels}
     print(response_dict)
     response_string = json.dumps(response_dict)
     return response_string