alihan/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2023-11-04 02:52:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Extend C-style API with full inference methods

Browse Source
This commit is contained in:
Georgi Gerganov
2022-10-04 22:43:37 +03:00
parent 6b77124e01
commit eba33adadd
4 changed files with 328 additions and 348 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
whisper.h
View File

@@ -1,6 +1,8 @@
#ifndef WHISPER_H
#define WHISPER_H
#include <stdint.h>
#ifdef WHISPER_SHARED
# ifdef _WIN32
# ifdef WHISPER_BUILD
@@ -15,6 +17,12 @@
# define WHISPER_API
#endif
#define WHISPER_SAMPLE_RATE 16000
#define WHISPER_N_FFT 400
#define WHISPER_N_MEL 80
#define WHISPER_HOP_LENGTH 160
#define WHISPER_CHUNK_SIZE 30
#ifdef __cplusplus
extern "C" {
#endif
@@ -23,12 +31,6 @@ extern "C" {
// C interface
//
#define SAMPLE_RATE 16000
#define N_FFT 400
#define N_MEL 80
#define HOP_LENGTH 160
#define CHUNK_SIZE 30
// TODO: documentation will come soon
struct whisper_context;
@@ -101,7 +103,9 @@ extern "C" {
int n_threads;
bool transcribe;
bool translate;
bool print_special_tokens;
bool print_progress;
const char * language;
@@ -118,14 +122,22 @@ extern "C" {
};
};
WHISPER_API struct whisper_full_params whisper_full_default_params(enum whisper_decode_strategy strategy);
// full whisper run - encode + decode
// TODO: implement
WHISPER_API int whisper_full(
struct whisper_context * ctx,
struct whisper_full_params * params,
struct whisper_full_params params,
const float * samples,
int n_samples);
WHISPER_API int whisper_full_n_segments(struct whisper_context * ctx);
WHISPER_API int64_t whisper_full_get_segment_t0(struct whisper_context * ctx, int i_segment);
WHISPER_API int64_t whisper_full_get_segment_t1(struct whisper_context * ctx, int i_segment);
WHISPER_API const char * whisper_full_get_segment_text(struct whisper_context * ctx, int i_segment);
#ifdef __cplusplus
}
#endif