Update app.py

app.py

@@ -110,7 +110,7 @@ with gr.Blocks(css=css) as demo:
 
         DIS_map = predict(np.array(image))
         DIS_map = (DIS_map - DIS_map.min()) / (DIS_map.max() - DIS_map.min()) * 255.0
-        matting = (DIS_map[...,None] / 255.0 * original_image)
+        matting = (DIS_map[...,None] / 255.0 * original_image) + (255-matting[...,None])
         return matting.astype('uint8')
 
     submit.click(on_submit, inputs=[input_image], outputs=dis_image_slider)