mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/imputnet/cobalt.git synced 2025-01-15 19:35:31 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

web/removebg: fix types (remove garbage)
Some checks are pending
CodeQL / Analyze (${{ matrix.language }}) (none, javascript-typescript) (push) Waiting to run
Details
Run tests / check lockfile correctness (push) Waiting to run
Details
Run tests / web sanity check (push) Waiting to run
Details
Run tests / api sanity check (push) Waiting to run
Details

Browse source
This commit is contained in:
wukko 2025-01-14 18:30:33 +06:00
parent d649a00718
commit 2f2d39dc4c
No known key found for this signature in database
GPG key ID: 3E30B3F26C7B4AA2
1 changed files with 5 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
web/src/lib/workers/removebg.ts
View file

@ -4,20 +4,10 @@ const models = {
light: {
id: "briaai/RMBG-1.4",
input: "input",
modelConfig: {
device: "wasm",
dtype: "fp32",
},
processorConfig: {},
},
heavy: {
id: "onnx-community/BiRefNet_lite",
input: "input_image",
modelConfig: {
device: "webgpu",
dtype: "fp16",
},
processorConfig: {},
}
}
@ -25,9 +15,12 @@ export const removeImageBackground = async (file: File) => {
const image = await RawImage.fromBlob(new Blob([file]));
const model_type = "light";
const model = await AutoModel.from_pretrained(models[model_type].id, models[model_type].modelConfig);
const model = await AutoModel.from_pretrained(models[model_type].id, {
device: "wasm",
dtype: "fp32",
});
const processor = await AutoProcessor.from_pretrained(models[model_type].id, models[model_type].processorConfig);
const processor = await AutoProcessor.from_pretrained(models[model_type].id, {});
if (model && processor) {
const { pixel_values } = await processor(image);