Update app.py

app.py

@@ -51,13 +51,13 @@ class TargetMaskConvertTransform(ItemTransform):
         #Convert to array
         mask = np.array(mask)
 
-        mask[(mask!=255) & (mask!=150) & (mask!=76) & (mask!=74) & (mask!=29) & (mask!=25)]=0
         mask[mask==255]=1
         mask[mask==150]=2
-        mask[mask==76]=4
-        mask[mask==74]=4
-        mask[mask==29]=3
-        mask[mask==25]=3
+        mask[mask==76]=3
+        mask[mask==74]=3
+        mask[mask==29]=4
+        mask[mask==25]=4
+        mask[(mask != 1) & (mask != 2) & (mask != 3) & (mask != 4)] = 0
 
         # Back to PILMask
         mask = PILMask.create(mask)