Update app.py

app.py

@@ -10,7 +10,7 @@ from io import BytesIO
 def get_segmentation_mask(image_url):
     client = Client("facebook/sapiens-seg")
     result = client.predict(image=handle_file(image_url), model_name="1b", api_name="/process_image")
-    return np.load(result[1])  # Result[2] contains the .npy mask
+    return np.load(result[1])  # Result[1] contains the .npy mask
 
 def process_image(image, categories_to_hide):
     # Convert uploaded image to a PIL Image
@@ -26,7 +26,7 @@ def process_image(image, categories_to_hide):
         "Clothes": [1, 12, 22, 8, 9, 17, 18],  # Includes Shoes, Socks, Slippers
         "Face": [2, 23, 24, 25, 26, 27],  # Face Neck, Lips, Teeth, Tongue
         "Hair": [3],  # Hair
-        "Skin (Hands, Feet, Body)": [4, 5, 6, 7, 10, 11, 13, 14, 15, 16, 19, 20, 21]  # Hands, Feet, Arms, Legs, Torso
+        "Skin": [4, 5, 6, 7, 10, 11, 13, 14, 15, 16, 19, 20, 21]  # Hands, Feet, Arms, Legs, Torso
     }
 
     # Convert image to numpy array (RGBA)
@@ -55,7 +55,7 @@ demo = gr.Interface(
     inputs=[
         gr.File(label="Upload an Image"),
         gr.CheckboxGroup([
-            "Background", "Clothes", "Face", "Hair", "Skin (Hands, Feet, Body)"
+            "Background", "Clothes", "Face", "Hair", "Skin"
         ], label="Select Categories to Preserve")
     ],
     outputs=gr.Image(label="Masked Image", type="pil"),