fix use after free

jomjol · Mar 17, 2023 · 1e6aa3f · 1e6aa3f
1 parent b81a2b1
commit 1e6aa3f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/code/components/jomjol_tfliteclass/CTfLiteClass.cpp b/code/components/jomjol_tfliteclass/CTfLiteClass.cpp
@@ -312,11 +312,11 @@ CTfLiteClass::CTfLiteClass()
 
 CTfLiteClass::~CTfLiteClass()
 {
-  free(modelfile);
-
-  free(this->tensor_arena);
   delete this->interpreter;
   delete this->error_reporter;
+
+  free_psram_heap(std::string(TAG) + "->modelfile", modelfile);
+  free_psram_heap(std::string(TAG) + "->tensor_arena", this->tensor_arena);
 }