Update handler.py
Browse files- handler.py +3 -9
handler.py
CHANGED
|
@@ -150,17 +150,11 @@ class EndpointHandler:
|
|
| 150 |
[(p.get("answer"), round(p.get("score", 0.0), 3)) for p in current_preds[:5]]
|
| 151 |
)
|
| 152 |
|
| 153 |
-
MIN_SPAN_SCORE = 0.25 # tune this
|
| 154 |
-
|
| 155 |
-
filtered_preds = [
|
| 156 |
-
p for p in current_preds
|
| 157 |
-
if p.get("score", 0.0) >= MIN_SPAN_SCORE
|
| 158 |
-
]
|
| 159 |
-
|
| 160 |
# Extract the answer strings
|
| 161 |
-
raw_answers = [p
|
| 162 |
|
| 163 |
-
|
|
|
|
| 164 |
|
| 165 |
results.append({
|
| 166 |
"utterance": utterance,
|
|
|
|
| 150 |
[(p.get("answer"), round(p.get("score", 0.0), 3)) for p in current_preds[:5]]
|
| 151 |
)
|
| 152 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 153 |
# Extract the answer strings
|
| 154 |
+
raw_answers = [p.get('answer', '') for p in current_preds]
|
| 155 |
|
| 156 |
+
# Clean spans using your original logic
|
| 157 |
+
triggers = self._clean_spans(raw_answers, utterance)
|
| 158 |
|
| 159 |
results.append({
|
| 160 |
"utterance": utterance,
|