We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bef12ff commit ccd9e84Copy full SHA for ccd9e84
extensions/firestore-huggingface-inference-api/functions/src/run_inference.ts
@@ -152,9 +152,13 @@ export async function runHostedInference(
152
model: config.modelId,
153
}),
154
inputs: inputs,
155
+ parameters: {
156
+ return_all_scores: true,
157
+ top_k: null,
158
+ },
159
};
160
- return await inference.textClassification(options);
161
+ return await inference.request<TextClassificationOutput>(options);
162
}
163
164
case Task.textGeneration:
0 commit comments