diff --git a/example/image_segmentation/pubspec.yaml b/example/image_segmentation/pubspec.yaml
index 9b0a9cd..731d2b5 100644
--- a/example/image_segmentation/pubspec.yaml
+++ b/example/image_segmentation/pubspec.yaml
@@ -64,8 +64,7 @@ flutter:
   # To add assets to your application, add an assets section, like this:
   assets:
     - assets/images/
-    - assets/deeplabv3.tflite
-    - assets/labelmap.txt
+    - assets/
 
   # An image asset can refer to one or more resolution-specific "variants", see
   # https://flutter.dev/assets-and-images/#resolution-aware