diff --git a/build/lotus-denoise.js b/build/lotus-denoise.js index 7266ecb8e..b82640067 100644 --- a/build/lotus-denoise.js +++ b/build/lotus-denoise.js @@ -54,14 +54,6 @@ script: 'speexWorklet.js', wasm: 'speex.wasm', }, - dtln: { - name: 'dtln-processor', - script: 'dtlnWorklet.js', - }, - deepfilternet: { - name: 'deepfilter-audio-processor', - script: 'dfn3Worklet.js', - }, gate: { name: '@sapphi-red/web-noise-suppressor/noise-gate', script: 'noiseGateWorklet.js', @@ -156,31 +148,14 @@ } // 2. ML Processor - var mlOptions = { + var mlNode = new AudioWorkletNode(ctx, PROCESSORS[MODEL].name, { channelCount: 1, + channelCountMode: 'explicit', numberOfInputs: 1, numberOfOutputs: 1, - processorOptions: { maxChannels: 1 } - }; - - if (MODEL === 'rnnoise' || MODEL === 'speex') { - mlOptions.processorOptions.wasmBinary = wasmBinary; - } else if (MODEL === 'dtln') { - mlOptions.processorOptions = { - wasmUrl: ASSET_BASE + 'litert_wasm_internal.wasm', - model1Url: ASSET_BASE + 'model_1.tflite', - model2Url: ASSET_BASE + 'model_2.tflite', - processorUrl: ASSET_BASE + 'dtlnProcessor.js' - }; - } else if (MODEL === 'deepfilternet') { - mlOptions.processorOptions = { - wasmModule: wasmBinary, - modelBytes: new Uint8Array(wasmBinary), - suppressionLevel: 50 - }; - } - - var mlNode = new AudioWorkletNode(ctx, PROCESSORS[MODEL].name, mlOptions); + outputChannelCount: [1], + processorOptions: { maxChannels: 1, wasmBinary: wasmBinary }, + }); head.connect(mlNode); mlNode.connect(dest);