import { type FaceDetectorOptions } from '@avatune/face-detector';
import type { SkinTonePredictorClass } from '@avatune/types';
import * as tf from '@tensorflow/tfjs';
type EthnicityClass = 'black' | 'east_asian' | 'indian' | 'latino_hispanic' | 'middle_eastern' | 'southeast_asian' | 'white';
export type SkinToneResult = {
    tone: SkinTonePredictorClass;
    confidence: number;
    probabilities: Record<SkinTonePredictorClass, number>;
    ethnicity: EthnicityClass;
    ethnicityConfidence: number;
    faceDetected: boolean;
};
export type SkinTonePredictorOptions = {
    modelDir: string;
    faceDetection?: boolean | FaceDetectorOptions;
};
export declare function createSkinTonePredictor(modelDir?: string): {
    loadModel(): Promise<void>;
    predict(imageTensor: tf.Tensor3D): Promise<SkinToneResult>;
    predictFromImage(image: HTMLImageElement | HTMLVideoElement | HTMLCanvasElement): Promise<SkinToneResult>;
};
export {};