Update app.py

app.py

@@ -50,10 +50,11 @@ def get_sample_images(image_folder, format=[('.jpg', '.jpeg'), ('.png')]):
     img_file_list.sort()
     
     # Filter out only the image files (assuming images have extensions like '.jpg')
-    image_files = [[image_folder +'/' + file] for file in img_file_list if file.lower().endswith(format[0])]
-    mask_files = [[image_folder +'/' + file] for file in img_file_list if file.lower().endswith(format[1])]
+    image_files = [image_folder +'/' + file for file in img_file_list if file.lower().endswith(format[0])]
+    mask_files = [image_folder +'/' + file for file in img_file_list if file.lower().endswith(format[1])]
 
     if mask_files == []:
+        image_files = [[image] for image in image_files]
         return image_files
 
     return [list(pair) for pair in zip(image_files, mask_files)]