diff --git a/src/app/features/settings/general/DenoiseTester.tsx b/src/app/features/settings/general/DenoiseTester.tsx index e3d48531f..acd8d511f 100644 --- a/src/app/features/settings/general/DenoiseTester.tsx +++ b/src/app/features/settings/general/DenoiseTester.tsx @@ -1,6 +1,7 @@ import React, { useCallback, useEffect, useRef, useState } from 'react'; import { Box, Button, Text } from 'folds'; import { DenoiseModelId } from '../../../state/settings'; +import { DENOISE_MODELS } from '../../../utils/lotusDenoiseUtils'; import { DenoiseNode, buildGateNode, @@ -307,7 +308,7 @@ export function DenoiseTester({ model, useGate, gateThreshold, nativeNS }: Denoi [stopLive, stopPlayback], ); - const modelLabel = model === 'rnnoise' ? 'RNNoise' : model === 'speex' ? 'Speex' : 'DTLN'; + const modelLabel = DENOISE_MODELS.find((m) => m.id === model)?.name ?? model; return ( @@ -367,6 +368,7 @@ export function DenoiseTester({ model, useGate, gateThreshold, nativeNS }: Denoi { label: 'RNNoise', model: 'rnnoise' }, { label: 'Speex', model: 'speex' }, { label: 'DTLN', model: 'dtln' }, + { label: 'DeepFilterNet', model: 'deepfilternet' }, ] as const ).map((b) => (