SetFit: Cast probabilities to float & increment version (#411)

tomaarsen · web-flow · commit 20510ecd2c1c · 2024-03-04T09:54:38.000+01:00
* Cast output values to float

* Increment SetFit version to most recent
diff --git a/docker_images/setfit/app/pipelines/text_classification.py b/docker_images/setfit/app/pipelines/text_classification.py
@@ -26,7 +26,7 @@ def __call__(self, inputs: str) -> List[Dict[str, float]]:
             id2label = getattr(self.model, "id2label", {}) or {}
             return [
                 [
-                    {"label": id2label.get(idx, idx), "score": prob}
+                    {"label": id2label.get(idx, idx), "score": float(prob)}
                     for idx, prob in enumerate(probs[0])
                 ]
             ]
diff --git a/docker_images/setfit/requirements.txt b/docker_images/setfit/requirements.txt
@@ -1,4 +1,4 @@
 starlette==0.27.0
 git+https://github.com/huggingface/api-inference-community.git@f06a71e72e92caeebabaeced979eacb3542bf2ca
 huggingface_hub==0.20.2
-setfit==1.0.1
+setfit==1.0.3

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ def __call__(self, inputs: str) -> List[Dict[str, float]]:`
`26`	`26`	`id2label = getattr(self.model, "id2label", {}) or {}`
`27`	`27`	`return [`
`28`	`28`	`[`
`29`		`- {"label": id2label.get(idx, idx), "score": prob}`
	`29`	`+ {"label": id2label.get(idx, idx), "score": float(prob)}`
`30`	`30`	`for idx, prob in enumerate(probs[0])`
`31`	`31`	`]`
`32`	`32`	`]`