Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -157,7 +157,7 @@ def predict(images, resolution, weights_file):
|
|
| 157 |
save_file_path = os.path.join(save_dir, "{}.png".format(os.path.splitext(os.path.basename(image_src))[0]))
|
| 158 |
image_masked.save(save_file_path)
|
| 159 |
save_paths.append(save_file_path)
|
| 160 |
-
image_masked =
|
| 161 |
|
| 162 |
if tab_is_batch:
|
| 163 |
zip_file_path = os.path.join(save_dir, "{}.zip".format(save_dir))
|
|
|
|
| 157 |
save_file_path = os.path.join(save_dir, "{}.png".format(os.path.splitext(os.path.basename(image_src))[0]))
|
| 158 |
image_masked.save(save_file_path)
|
| 159 |
save_paths.append(save_file_path)
|
| 160 |
+
image_masked = np.array(image_masked.convert('RGB'))
|
| 161 |
|
| 162 |
if tab_is_batch:
|
| 163 |
zip_file_path = os.path.join(save_dir, "{}.zip".format(save_dir))
|