diff options
Diffstat (limited to 'applications/inference_process/src/inference_process.cpp')
-rw-r--r-- | applications/inference_process/src/inference_process.cpp | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/applications/inference_process/src/inference_process.cpp b/applications/inference_process/src/inference_process.cpp index 29254c7..e96d601 100644 --- a/applications/inference_process/src/inference_process.cpp +++ b/applications/inference_process/src/inference_process.cpp @@ -119,11 +119,9 @@ bool InferenceProcess::runJob(InferenceJob &job) { RegisterDebugLogCallback(tfluDebugLog); // Get model handle and verify that the version is correct - const tflite::Model *model = ::tflite::GetModel(job.networkModel.data); - if (model->version() != TFLITE_SCHEMA_VERSION) { - LOG_ERR("Model schema version unsupported: version=%" PRIu32 ", supported=%d.", - model->version(), - TFLITE_SCHEMA_VERSION); + const tflite::Model *model = parser.getModel(job.networkModel.data, job.networkModel.size); + if (model == nullptr) { + LOG_ERR("Invalid model"); return true; } |