Update app.py

app.py

@@ -23,6 +23,7 @@ transform_image = transforms.Compose([
     transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
 ])
 
+@spaces.GPU
 def process(image: Image.Image) -> Image.Image:
     image_size = image.size
     input_tensor = transform_image(image).unsqueeze(0).to(device)
@@ -38,6 +39,7 @@ def process(image: Image.Image) -> Image.Image:
     result = Image.composite(image, white_bg, binary_mask)
     return result
 
+@spaces.GPU
 def handler(image=None, image_url=None, batch_urls=None) -> Union[str, List[str], None]:
     results = []