Added comment and changed colors

app.py

@@ -99,9 +99,9 @@ def segment_images(model_name,url_input,image_input,threshold):
 
         r_shape = results["segmentation"].shape
         new_image = np.zeros((r_shape[0], r_shape[1], 3), dtype=np.uint8)
-        new_image[:, :, 0] = results["segmentation"].numpy()[:, :]
-        new_image[:, :, 1] = (new_image[:, :, 0] * 2) %256
-        new_image[:, :, 2] = (new_image[:, :, 0] * 3) %256
+        new_image[:, :, 0] = (results["segmentation"].numpy()[:, :] * 2) % 256
+        new_image[:, :, 1] = (new_image[:, :, 0] * 3) %256
+        new_image[:, :, 2] = (new_image[:, :, 0] * 4) %256
 
         new_image = Image.fromarray(new_image)
 
@@ -126,6 +126,8 @@ Links to HuggingFace Models:
 - [facebook/detr-resnet-50-panoptic](https://huggingface.co/facebook/detr-resnet-50-panoptic)   (Not implemented YET)
 - [facebook/detr-resnet-101-panoptic](https://huggingface.co/facebook/detr-resnet-101-panoptic) (Not implemented YET)
 - [facebook/maskformer-swin-large-coco](https://huggingface.co/facebook/maskformer-swin-large-coco)
+
+Warning: On the free tier, MaskFormer takes a long time. 
 """
 
 models = ["facebook/detr-resnet-50-panoptic","facebook/detr-resnet-101-panoptic","facebook/maskformer-swin-large-coco"]