Spaces:
Running
Running
Commit
•
6896fb7
1
Parent(s):
b6bb35e
Show image in output
Browse files
app.py
CHANGED
@@ -210,6 +210,7 @@ def refresh_predictions(predictor, input_img, output_img, click_coords, click_la
|
|
210 |
|
211 |
out_viz = [
|
212 |
viz_pred_mask(input_img, mask_to_viz, point_coords=None, point_labels=None, bbox_coords=None, seperate_scribble_masks=None, binary=binary_checkbox),
|
|
|
213 |
255*(mask_to_viz[...,None].repeat(axis=2, repeats=3)>0.5) if binary_checkbox else mask_to_viz[...,None].repeat(axis=2, repeats=3),
|
214 |
]
|
215 |
|
|
|
210 |
|
211 |
out_viz = [
|
212 |
viz_pred_mask(input_img, mask_to_viz, point_coords=None, point_labels=None, bbox_coords=None, seperate_scribble_masks=None, binary=binary_checkbox),
|
213 |
+
input_img,
|
214 |
255*(mask_to_viz[...,None].repeat(axis=2, repeats=3)>0.5) if binary_checkbox else mask_to_viz[...,None].repeat(axis=2, repeats=3),
|
215 |
]
|
216 |
|