Update app.py

app.py

@@ -10,22 +10,18 @@ import time
 import threading
 import argparse
 
-
-
 def list_iter_images(save_dir):
-    # Specify the image extensions you want to search for
-    image_extensions = ['jpg', 'jpeg', 'png', 'gif', 'bmp']  # Add more if needed
+    # Specify only PNG images
+    image_extension = 'png'
 
     # Create a list to store the image file paths
     image_paths = []
 
-    # Iterate through the specified image extensions and get the file paths
-    for ext in image_extensions:
-        # Use glob to find all image files with the given extension
-        image_paths.extend(glob.glob(os.path.join(save_dir, f'*.{ext}')))
+    # Use glob to find all PNG image files
+    all_images = glob.glob(os.path.join(save_dir, f'*.{image_extension}'))
 
-    # Now image_paths contains the list of all image file paths
-    #print(image_paths)
+    # Filter out 'best_image.png'
+    image_paths = [img for img in all_images if os.path.basename(img) != 'best_image.png']
 
     return image_paths