fix bug with images

app.py

@@ -19,6 +19,9 @@ from torchvision.transforms.functional import to_pil_image
 from tqdm import tqdm
 from util import transform
 
+# print the current working directory
+print(os.getcwd())
+
 IMAGE_PATH = 'src\\examples'
 RANDOM_IMAGES_TO_SHOW = 20
 IMAGES_PER_ROW = 5
@@ -162,6 +165,7 @@ with gr.Blocks() as demo:
             showed_images = list()
             loaded_images = load_random_images()
             amount_rows = max(1, (len(loaded_images) // IMAGES_PER_ROW))
+            if len(loaded_images) == 0: amount_rows = 0
             for i in range(amount_rows):
                 with gr.Row():
                     for j in range(IMAGES_PER_ROW):