Spaces:
Running
on
Zero
Running
on
Zero
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))
|