Experimenting with few-shot classification.

app.py

@@ -119,7 +119,7 @@ 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}
+    response_dict = {'scores': scores, 'labels': data['candidate_labels']}
     print(response_dict)
     response_string = json.dumps(response_dict)
     return response_string