web/removebg: differentiate messaging even more, add temporary logging

This commit is contained in:
wukko
2025-01-17 01:45:11 +06:00
parent cc3e3be118
commit b85771dc1d
2 changed files with 26 additions and 14 deletions

View File

@@ -40,24 +40,31 @@ const removeImageBackground = async (file: File) => {
dtype: "fp32",
});
console.log("we're past model loading!");
const processor = await AutoProcessor.from_pretrained(models[model_type].id, {});
console.log("now also past processor!");
if (model && processor) {
const { pixel_values } = await processor(image);
console.log("got pixel values");
const { output } = await model({ [models[model_type].input]: pixel_values });
console.log("got output");
const mask = await RawImage.fromTensor(output[0].mul(255).to('uint8')).resize(image.width, image.height);
console.log("got the mask");
self.postMessage({
cobaltRemoveBgWorker: {
result: maskImage(image, mask),
}
});
});
}
}
self.onmessage = async (event: MessageEvent) => {
if (event.data.file) {
await removeImageBackground(event.data.file);
if (event.data.cobaltRemoveBgWorker.file) {
await removeImageBackground(event.data.cobaltRemoveBgWorker.file);
self.close();
}
self.close();
}