diff --git a/include/rnnoise.h b/include/rnnoise.h index 511a9768..f7f5007a 100644 --- a/include/rnnoise.h +++ b/include/rnnoise.h @@ -53,6 +53,8 @@ typedef struct RNNModel RNNModel; RNNOISE_EXPORT int rnnoise_get_size(); +RNNOISE_EXPORT int rnnoise_get_frame_size(); + RNNOISE_EXPORT int rnnoise_init(DenoiseState *st, RNNModel *model); RNNOISE_EXPORT DenoiseState *rnnoise_create(RNNModel *model); diff --git a/src/denoise.c b/src/denoise.c index d1c21dc0..5a628440 100644 --- a/src/denoise.c +++ b/src/denoise.c @@ -257,6 +257,10 @@ int rnnoise_get_size() { return sizeof(DenoiseState); } +int rnnoise_get_frame_size() { + return FRAME_SIZE; +} + int rnnoise_init(DenoiseState *st, RNNModel *model) { memset(st, 0, sizeof(*st)); if (model)