Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def get_sample_images(image_folder):
|
|
42 |
img_file_list = os.listdir(image_folder)
|
43 |
|
44 |
# Filter out only the image files (assuming images have extensions like '.jpg'.)
|
45 |
-
image_files = [[file] for file in img_file_list if file.lower().endswith(('.jpg', '.jpeg'))]
|
46 |
|
47 |
return image_files
|
48 |
|
|
|
42 |
img_file_list = os.listdir(image_folder)
|
43 |
|
44 |
# Filter out only the image files (assuming images have extensions like '.jpg'.)
|
45 |
+
image_files = [[image_folder +'/' + file] for file in img_file_list if file.lower().endswith(('.jpg', '.jpeg'))]
|
46 |
|
47 |
return image_files
|
48 |
|