diff --git a/demos/tensorflow_lite/run_tflite_in_occlum.sh b/demos/tensorflow_lite/run_tflite_in_occlum.sh
index a87f1e7e..bc4f6d70 100755
--- a/demos/tensorflow_lite/run_tflite_in_occlum.sh
+++ b/demos/tensorflow_lite/run_tflite_in_occlum.sh
@@ -15,10 +15,8 @@ init_workspace() {
 }
 
 build_occlum() {
-    cp ../tensorflow_src/tensorflow/lite/tools/make/gen/linux_x86_64/bin/* image/bin
-    cp /usr/local/occlum/x86_64-linux-musl/lib/libz.so.1 image/lib
-    cp -r ../models image
-    cp -r ../tensorflow_src/tensorflow/lite/examples/label_image/testdata image
+    rm -rf image
+    copy_bom -f ../tflite.yaml --root image --include-dir /opt/occlum/etc/template;
     occlum build
 }
 
diff --git a/demos/tensorflow_lite/tflite.yaml b/demos/tensorflow_lite/tflite.yaml
new file mode 100644
index 00000000..9799ba30
--- /dev/null
+++ b/demos/tensorflow_lite/tflite.yaml
@@ -0,0 +1,9 @@
+includes:
+  - base.yaml
+targets:
+  - target: /
+    copy:
+      - dirs:
+        - ../tensorflow_src/tensorflow/lite/tools/make/gen/linux_x86_64/bin
+        - ../models
+        - ../tensorflow_src/tensorflow/lite/examples/label_image/testdata