Khriis commited on
Commit
b8c5ac6
verified
1 Parent(s): 4c765f3

Update handler.py

Browse files
Files changed (1) hide show
  1. 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["answer"] for p in filtered_preds]
162
 
163
- triggers = self._clean_spans(raw_answers, utterance)
 
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,