Update app.py

app.py

@@ -53,8 +53,8 @@ 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 mask_files == []:
-        image_files = [[image] for image in image_files]
+    if not mask_files:
+        image_files = [[image_folder +'/' + file] for file in img_file_list if file.lower().endswith(format[0])]
         return image_files
 
     return [list(pair) for pair in zip(image_files, mask_files)]
@@ -200,7 +200,7 @@ tab1 = gr.Interface(
     outputs=[gr.Image(label='Ground Truth'), gr.Image(label='Predicted')],
     title='Images with Ground Truth',
     description=description,
-    examples=train_images[:5]
+    examples=train_images
 )
 
 # Create the test dataset interface
@@ -210,7 +210,7 @@ tab2 = gr.Interface(
     outputs=gr.Image(label='Predicted'),
     title='Images without Ground Truth',
     description=description,
-    examples=test_images[:5]
+    examples=test_images
 )
 
 # Create a Multi Interface with Tabs