fix gpu error

app.py

@@ -107,7 +107,7 @@ def process_image(args, inpainting_pipeline, net):
                                                     num_inference_steps=args.num_steps).images[0]
     clothed_image_from_pipeline = remove(clothed_image_from_pipeline)
     clothed_image_from_pipeline = change_bg_color(clothed_image_from_pipeline, "WHITE")
-    return clothed_image_from_pipeline.convert("RGB")
+    return clothed_image_from_pipeline.convert("RGB"). mask_PIL
 
 net = load_u2net()
 inpainting_pipeline = load_inpainting_pipeline()
@@ -141,8 +141,9 @@ stable_fashion_args.num_steps = num_steps
 
 
 if file_name is not None:
-    result_image = process_image(stable_fashion_args, inpainting_pipeline, net)
+    result_image, mask_PIL = process_image(stable_fashion_args, inpainting_pipeline, net)
     st.image(result_image, caption='Result')
+    st.image(mask_PIL, caption='Mask')
 else:
     stock_image = Image.open('assets/abhishek_yellow.jpg')
     st.image(stock_image, caption='Result')