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[2])  # Result[2] contains the .npy mask
+    return np.load(result[1])  # Result[2] contains the .npy mask
 
 def process_image(image, categories_to_hide):
     # Convert uploaded image to a PIL Image