diff options
| -rw-r--r-- | include/rnnoise.h | 2 | ||||
| -rw-r--r-- | src/denoise.c | 4 |
2 files changed, 6 insertions, 0 deletions
diff --git a/include/rnnoise.h b/include/rnnoise.h index 511a976..f7f5007 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 d1c21dc..5a62844 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) |