Run RMBG-1.4 in your browser

import { AutoModel, AutoProcessor, RawImage } from "@huggingface/transformers";

// RMBG-1.4's config uses a custom architecture string, so pipeline() rejects
// it; load it as a custom model with an explicit processor config instead
// (this mirrors the official remove-background-web demo).
const model = await AutoModel.from_pretrained("briaai/RMBG-1.4", {
  config: { model_type: "custom" },
  device: "webgpu", // or "wasm"
  dtype: "fp16", // use "q8" for the WASM build
});
const processor = await AutoProcessor.from_pretrained("briaai/RMBG-1.4", {
  config: {
    do_normalize: true, do_pad: false, do_rescale: true, do_resize: true,
    image_mean: [0.5, 0.5, 0.5], image_std: [1, 1, 1],
    feature_extractor_type: "ImageFeatureExtractor",
    resample: 2, rescale_factor: 1 / 255, size: { width: 1024, height: 1024 },
  },
});

const image = await RawImage.fromURL("https://your-image-url.jpg");
const { pixel_values } = await processor(image);
const { output } = await model({ input: pixel_values });
// output[0] is the alpha mask; resize it to the source and composite.
const mask = await RawImage.fromTensor(output[0].mul(255).to("uint8"))
  .resize(image.width, image.height);