1 files changed, 2 insertions, 6 deletions
diff --git a/applications/baremetal/main.cpp b/applications/baremetal/main.cpp
index e9b398a..ab365c6 100644
--- a/applications/baremetal/main.cpp
+++ b/applications/baremetal/main.cpp
@@ -45,13 +45,9 @@ using namespace InferenceProcess;
  * InferenceJob
  ****************************************************************************/
 
-#ifndef TENSOR_ARENA_SIZE
-#define TENSOR_ARENA_SIZE 2000000
-#endif
-
-__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorArena[TENSOR_ARENA_SIZE];
+__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorArena[tensorArenaSize];
 
-class InferenceProcess inferenceProcess(TFLuTensorArena, TENSOR_ARENA_SIZE);
+class InferenceProcess inferenceProcess(TFLuTensorArena, tensorArenaSize);
 
 uint8_t outputData[sizeof(expectedOutputData)] __attribute__((aligned(16), section("output_data_sec")));