Update app.py

app.py

@@ -79,7 +79,7 @@ def generate_agnostic(image: Image.Image, segmentation):
     img_np = np.array(image.resize((192, 256)))
     agnostic_np = img_np.copy()
     segmentation_resized = cv2.resize(segmentation.astype(np.uint8), (192, 256), interpolation=cv2.INTER_NEAREST)
-    clothing_labels = [4, 5, 6, 7, 8, 16]
+    clothing_labels = [4]
     for label in clothing_labels:
         agnostic_np[segmentation_resized == label] = [128, 128, 128]
     return Image.fromarray(agnostic_np)