hookformers
Version:
Simple, performant React Hooks for running Transformers.js in your browser.
5 lines • 445 B
TypeScript
import { PretrainedModelOptions, TextClassificationOutput, TextClassificationPipelineOptions } from "@huggingface/transformers";
export declare function useTextClassification(model: string, options?: PretrainedModelOptions): {
predict: (input: string | string[], options?: TextClassificationPipelineOptions | undefined) => Promise<TextClassificationOutput | TextClassificationOutput[]>;
};
//# sourceMappingURL=useTextClassification.d.ts.map