#ifndef MAIN_H
#define MAIN_H

#ifdef __cplusplus
extern "C" {
#endif

int generate(const char *model_path, const char *prompt);

#ifdef __cplusplus
}
#endif

#endif // MAIN_H