adrienbrdne commited on
Commit
f423824
·
verified ·
1 Parent(s): 58d1f23

Update scoring/specificity.py

Browse files
Files changed (1) hide show
  1. scoring/specificity.py +2 -2
scoring/specificity.py CHANGED
@@ -20,7 +20,7 @@ class PredictionsResponse(BaseModel):
20
  results: List[Dict[str, Union[str, float]]]
21
 
22
  class BatchPredictionScoreItem(BaseModel):
23
- problem_description: str
24
  score: float
25
 
26
  # Model environment variables
@@ -114,7 +114,7 @@ def predict_batch(items: ProblematicList):
114
 
115
  results.append(
116
  BatchPredictionScoreItem(
117
- problem_description=batch_texts[j],
118
  score=score_class_1
119
  )
120
  )
 
20
  results: List[Dict[str, Union[str, float]]]
21
 
22
  class BatchPredictionScoreItem(BaseModel):
23
+ problematic: str
24
  score: float
25
 
26
  # Model environment variables
 
114
 
115
  results.append(
116
  BatchPredictionScoreItem(
117
+ problematic=batch_texts[j],
118
  score=score_class_1
119
  )
120
  )