Add fast-foreground-estimation in masking image.

app.py

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