Lord-Raven commited on
Commit
a22030c
·
1 Parent(s): 46aa75d

Experimenting with few-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def zero_shot_classification(data):
75
 
76
  def few_shot_classification(data):
77
  results = onnx_few_shot_model(data['sequence'])
78
- response_string = json.dumps(results)
79
  return response_string
80
 
81
  gradio_interface = gradio.Interface(
 
75
 
76
  def few_shot_classification(data):
77
  results = onnx_few_shot_model(data['sequence'])
78
+ response_string = json.dumps(results.toList)
79
  return response_string
80
 
81
  gradio_interface = gradio.Interface(