Update app.py

app.py

@@ -42,7 +42,7 @@ def process(img):
     path = load_img(img,output_type="str")
     img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)
     mask = get_mask(model, img, use_amp= False, s=img_size)
-    img = mask * img
+    img = np.concatenate((img, mask * img, mask.repeat(3, 2) * 255), axis=1).astype(np.uint8)
     out = load_img(img)
     return out