Show image in output

app.py

@@ -210,6 +210,7 @@ def refresh_predictions(predictor, input_img, output_img, click_coords, click_la
     
     out_viz = [
         viz_pred_mask(input_img, mask_to_viz, point_coords=None, point_labels=None, bbox_coords=None, seperate_scribble_masks=None, binary=binary_checkbox),
+        input_img,
         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),
     ]