Update app.py

app.py

@@ -21,7 +21,7 @@ transform_image = transforms.Compose(
     ]
 )
 
-def fn(image: Union[PILImage.Image, str]) -> Tuple[PILImage.Image, PILImage.Image]:
+def fn(image: Union[Image.Image, str]) -> Tuple[Image.Image, Image.Image]:
     """
     Remove the background from an image and return both the transparent version and the original.
 
@@ -44,7 +44,7 @@ def fn(image: Union[PILImage.Image, str]) -> Tuple[PILImage.Image, PILImage.Imag
     return (processed_image, origin)
 
 @spaces.GPU
-def process(image: PILImage.Image) -> PILImage.Image:
+def process(image: Image.Image) -> Image.Image:
     """
     Apply BiRefNet-based image segmentation to remove the background.