Update app.py

app.py

@@ -10,6 +10,7 @@ from SegCloth import segment_clothing
 from transparent_background import Remover
 import threading
 import logging
+import spaces
 
 app = Flask(__name__)
 
@@ -45,7 +46,8 @@ def encode_image_to_base64(image):
     buffered = BytesIO()
     image.save(buffered, format="PNG")  # Use PNG for compatibility with RGBA
     return base64.b64encode(buffered.getvalue()).decode('utf-8')
-
+    
+@spaces.GPU
 def remove_background(image):
     remover = Remover()
     if isinstance(image, Image.Image):