We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a27b5b4 commit 7768e88Copy full SHA for 7768e88
packages/inference/src/providers/hf-inference.ts
@@ -303,7 +303,10 @@ export class HFInferenceImageSegmentationTask extends HFInferenceTask implements
303
override async getResponse(response: ImageSegmentationOutput): Promise<ImageSegmentationOutput> {
304
if (
305
Array.isArray(response) &&
306
- response.every((x) => typeof x.label === "string" && typeof x.mask === "string" && typeof x.score === "number")
+ response.every(
307
+ (x) =>
308
+ typeof x.label === "string" && typeof x.mask === "string" && (!("score" in x) || typeof x.score === "number")
309
+ )
310
) {
311
return response;
312
}
0 commit comments