Update app.py

app.py

@@ -48,8 +48,9 @@ def inference_instance(image):
     instance_seg_mask = result["segmentation"].cpu().detach().numpy()
 
     instance_seg_mask_disp = visualize_instance_seg_mask(instance_seg_mask)
-    image.paste(instance_seg_mask_disp, (0, 0), instance_seg_mask_disp)
-    st.image(image, caption='Result')
+    return instance_seg_mask_disp
 
 image = Image.open(file).convert("RGB")
-inference_instance(image)
+mask = inference_instance(image)
+image.paste(mask, (0, 0), mask)
+st.image(image, caption='Result')