amosfang commited on
Commit
de931c0
·
verified ·
1 Parent(s): 140f82a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = [[image] for image in 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)]