@@ -25,6 +25,6 @@ class SimpleSession(BaseSession):
pred = np.squeeze(pred)
mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
- mask = mask.resize(img.size, Image.Resampling.LANCZOS)
+ mask = mask.resize(img.size, Image.LANCZOS)
return [mask]