Update handler.py
Browse files- handler.py +3 -3
handler.py
CHANGED
@@ -24,9 +24,9 @@ class EndpointHandler():
|
|
24 |
- "label": A string representing what the label/class is. There can be multiple labels.
|
25 |
- "score": A score between 0 and 1 describing how confident the model is for this label/class.
|
26 |
"""
|
27 |
-
text = data.pop("text",
|
28 |
-
label_tolerance = data.pop("label_tolerance", 0)
|
29 |
-
backup_tolerance = data.pop("backup_tolerance", None)
|
30 |
|
31 |
# Return labeled results and backup results based on tolerances
|
32 |
inputs = self.preprocess_text(text)
|
|
|
24 |
- "label": A string representing what the label/class is. There can be multiple labels.
|
25 |
- "score": A score between 0 and 1 describing how confident the model is for this label/class.
|
26 |
"""
|
27 |
+
text = data['inputs'].pop("text", "")
|
28 |
+
label_tolerance = data['inputs'].pop("label_tolerance", 0)
|
29 |
+
backup_tolerance = data['inputs'].pop("backup_tolerance", None)
|
30 |
|
31 |
# Return labeled results and backup results based on tolerances
|
32 |
inputs = self.preprocess_text(text)
|