diff --git a/lib/galleryimage.dart b/lib/galleryimage.dart
index 7c8c3c2..f0106f3 100644
--- a/lib/galleryimage.dart
+++ b/lib/galleryimage.dart
@@ -61,12 +61,19 @@ class GalleryImage extends StatefulWidget {
 
 class _GalleryImageState extends State<GalleryImage> {
   List<GalleryItemModel> galleryItems = <GalleryItemModel>[];
+
   @override
   void initState() {
     _buildItemsList(widget.imageUrls);
     super.initState();
   }
 
+  @override
+  void didUpdateWidget(covariant GalleryImage oldWidget) {
+    _buildItemsList(widget.imageUrls);
+    super.didUpdateWidget(oldWidget);
+  }
+
   @override
   Widget build(BuildContext context) {
     return galleryItems.isEmpty