Lord-Raven commited on
Commit
7f947b9
·
1 Parent(s): fbcdba4

Experimenting with few-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -104,7 +104,7 @@ def zero_shot_classification(data):
104
 
105
  def few_shot_classification(data):
106
  results = onnx_few_shot_model(data['sequence'])
107
- print([classes[idx] for idx in results)
108
  response_string = json.dumps(results.tolist())
109
  return response_string
110
 
 
104
 
105
  def few_shot_classification(data):
106
  results = onnx_few_shot_model(data['sequence'])
107
+ print([classes[idx] for idx in results])
108
  response_string = json.dumps(results.tolist())
109
  return response_string
110