diff --git a/src/suite.py b/src/suite.py index 7538dab..4180f24 100644 --- a/src/suite.py +++ b/src/suite.py @@ -276,6 +276,9 @@ class MainApp: images = [] for file in glob.glob(path + "/*.jpg"): images.append(file) + + for file in glob.glob(path + "/*.png"): + images.append(file) self.img_max = len(images) self.img_name = os.path.split(images[self.img_current])