Update app.py

app.py

@@ -39,6 +39,18 @@ def encode_image_to_base64(image):
     image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
+#@spaces.GPU
+def remove_background(image):
+    remover = Remover()
+    if isinstance(image, Image.Image):
+        output = remover.process(image)
+    elif isinstance(image, np.ndarray):
+        image_pil = Image.fromarray(image)
+        output = remover.process(image_pil)
+    else:
+        raise TypeError("Unsupported image type")
+    return output
+
 # Détecter les personnes et segmenter leurs vêtements
 def detect_and_segment_persons(image, clothes):
     img = np.array(image)
@@ -76,18 +88,6 @@ def detect_and_segment_persons(image, clothes):
 
     return all_segmented_images
 
-    #@spaces.GPU
-    def remove_background(image):
-        remover = Remover()
-        if isinstance(image, Image.Image):
-            output = remover.process(image)
-        elif isinstance(image, np.ndarray):
-            image_pil = Image.fromarray(image)
-            output = remover.process(image_pil)
-        else:
-            raise TypeError("Unsupported image type")
-        return output
-
 @app.route('/api/detect', methods=['POST'])
 def detect():
     try: