Update app.py

app.py

@@ -46,6 +46,7 @@ def save_image(img):
     img.save(unique_name)
     return unique_name
 
+@spaces.GPU
 def rm_background(image):
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")
@@ -62,6 +63,7 @@ def rm_background(image):
     image.putalpha(mask)
     return image
 
+@spaces.GPU
 def detect_and_segment_persons(image, clothes):
     img = np.array(image)
     img = img[:, :, ::-1]  # RGB -> BGR
@@ -93,6 +95,7 @@ def detect_and_segment_persons(image, clothes):
 
     return all_segmented_images
 
+@spaces.GPU
 def process_image(input_image):
     try:
         clothes = ["Upper-clothes", "Skirt", "Pants", "Dress"]