fix bug with images

app.py

@@ -19,8 +19,7 @@ from torchvision.transforms.functional import to_pil_image
 from tqdm import tqdm
 from util import transform
 
-IMAGE_PATH = '/home/user/app/src/examples'
-print(os.listdir(IMAGE_PATH))
+IMAGE_PATH = os.path.join(os.getcwd(), 'src/examples')
 RANDOM_IMAGES_TO_SHOW = 20
 IMAGES_PER_ROW = 5
 
@@ -163,7 +162,9 @@ 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
+            if len(loaded_images) == 0:
+                print(f"Could not find any images in {IMAGE_PATH}")
+                amount_rows = 0
             for i in range(amount_rows):
                 with gr.Row():
                     for j in range(IMAGES_PER_ROW):