diff --git a/core/iwasm/libraries/wasi-nn/wasi_nn.c b/core/iwasm/libraries/wasi-nn/wasi_nn.c index 67e4cfc5a..1fbcdc46e 100644 --- a/core/iwasm/libraries/wasi-nn/wasi_nn.c +++ b/core/iwasm/libraries/wasi-nn/wasi_nn.c @@ -13,22 +13,7 @@ std::unique_ptr interpreter = NULL; std::unique_ptr model =NULL; -//Load model functions - -int main() -{ - char* f=read_file("ssd_tflite"); - - uint8_t *builder = (uint8_t*) f; - - uint8_t *builder_array[1] = { builder }; - - load(builder_array, 1); - - load(model builder, graph_encoding encoding) -} - - +ยบ void load(graph_builder_array builder, graph_encoding encoding) { model = tflite::FlatBufferModel::BuildFromBuffer(builder[0]);