Skip to content
This repository was archived by the owner on Jun 19, 2025. It is now read-only.

Commit 4726e4a

Browse files
committed
Loading tflite buffer from buffer working
1 parent b6dbd09 commit 4726e4a

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

native_client/tflitemodelstate.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,10 @@ TFLiteModelState::init(const std::string &model_string, bool init_from_bytes)
164164
}
165165

166166
if (init_from_bytes){
167-
fbmodel_ = tflite::FlatBufferModel::BuildFromBuffer(model_string.c_str(), model_string.size_t);
167+
char *tmp_buffer = new char[model_string.size()];
168+
std::copy(model_string.begin(), model_string.end(), tmp_buffer);
169+
// Using c_str does not work
170+
fbmodel_ = tflite::FlatBufferModel::VerifyAndBuildFromBuffer(tmp_buffer,model_string.size());
168171
if (!fbmodel_) {
169172
std::cerr << "Error at reading model buffer " << std::endl;
170173
return DS_ERR_FAIL_INIT_MMAP;
@@ -177,7 +180,6 @@ TFLiteModelState::init(const std::string &model_string, bool init_from_bytes)
177180
}
178181
}
179182

180-
181183
tflite::ops::builtin::BuiltinOpResolver resolver;
182184
tflite::InterpreterBuilder(*fbmodel_, resolver)(&interpreter_);
183185
if (!interpreter_) {

0 commit comments

Comments
 (0)