Lord-Raven commited on
Commit
d07a287
·
1 Parent(s): 2f56b86

Experimenting with few-shot classification.

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