Update app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def get_sample_images(image_folder, format=[('.jpg', '.jpeg'), ('.png')]):
|
|
| 54 |
mask_files = [image_folder +'/' + file for file in img_file_list if file.lower().endswith(format[1])]
|
| 55 |
|
| 56 |
if mask_files == []:
|
| 57 |
-
image_files = [
|
| 58 |
return image_files
|
| 59 |
|
| 60 |
return [list(pair) for pair in zip(image_files, mask_files)]
|
|
|
|
| 54 |
mask_files = [image_folder +'/' + file for file in img_file_list if file.lower().endswith(format[1])]
|
| 55 |
|
| 56 |
if mask_files == []:
|
| 57 |
+
image_files = [list(file) for file in image_files]
|
| 58 |
return image_files
|
| 59 |
|
| 60 |
return [list(pair) for pair in zip(image_files, mask_files)]
|