Update app.py

app.py

@@ -45,6 +45,10 @@ def load_model_file(filename):
 
 def get_sample_images(image_folder, format=[('.jpg', '.jpeg'), ('.png')]):
 
+    # Initialization
+    image_files = []
+    mask_files = []
+    
     # Get a list of all files in the folder
     img_file_list = os.listdir(image_folder)
     img_file_list.sort()
@@ -53,8 +57,9 @@ def get_sample_images(image_folder, format=[('.jpg', '.jpeg'), ('.png')]):
     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 not mask_files:
-        image_files = [[image_folder +'/' + file] for file in img_file_list if file.lower().endswith(format[0])]
+    if mask_files == []:
+        print(image_files)
+        image_files = [[file] for file in image_files]
         return image_files
 
     return [list(pair) for pair in zip(image_files, mask_files)]