Lord-Raven commited on
Commit
5a3b82a
·
1 Parent(s): cdb9220

Experimenting with few-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def few_shot_classification(data):
120
  probs = onnx_few_shot_model.predict_proba(sequences)
121
  scores = [true[0] for true in probs]
122
  response_dict = {'scores': scores, 'labels': data.candidate_labels}
123
- print()response_dict)
124
  response_string = json.dumps(response_dict)
125
  return response_string
126
 
 
120
  probs = onnx_few_shot_model.predict_proba(sequences)
121
  scores = [true[0] for true in probs]
122
  response_dict = {'scores': scores, 'labels': data.candidate_labels}
123
+ print(response_dict)
124
  response_string = json.dumps(response_dict)
125
  return response_string
126