Add fast-foreground-estimation in masking image.

app.py

@@ -148,7 +148,7 @@ def predict(images, resolution, weights_file):
 
         # Show Results
         pred_pil = transforms.ToPILImage()(pred)
-        image_masked = refine_foreground(image, pred_pil)
+        image_masked = refine_foreground(Image.fromarray(image), pred_pil)
         image_masked.putalpha(pred_pil.resize(image.size))
 
         torch.cuda.empty_cache()