Update app.py
Browse files
app.py
CHANGED
|
@@ -56,7 +56,7 @@ def get_sample_images(image_folder, format=[('.jpg', '.jpeg'), ('.png')]):
|
|
| 56 |
if mask_files == []:
|
| 57 |
return image_files
|
| 58 |
|
| 59 |
-
return image_files, mask_files
|
| 60 |
|
| 61 |
def ensemble_predict(X_array):
|
| 62 |
#
|
|
|
|
| 56 |
if mask_files == []:
|
| 57 |
return image_files
|
| 58 |
|
| 59 |
+
return [list(pair) for pair in zip(image_files, mask_files)]
|
| 60 |
|
| 61 |
def ensemble_predict(X_array):
|
| 62 |
#
|