Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
5a3b82a
1
Parent(s):
cdb9220
Experimenting with few-shot classification.
Browse files
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(
|
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 |
|