added seaborn

app.py

@@ -9,6 +9,8 @@ https://colab.research.google.com/github/NielsRogge/Transformers-Tutorials/blob/
 
 https://arxiv.org/abs/2005.12872
 
+https://arxiv.org/pdf/1801.00868.pdf 
+
 Additions
 - add shown labels as strings
 - show only animal masks (ask an nlp model?)
@@ -19,6 +21,7 @@ For next time
 - Im getting core dumped (segmentation fault) when loading hugging face model.. :()
     https://github.com/huggingface/transformers/issues/16939 
 - cap slider to 95?
+- switch between panoptic and semantic?
 """
 
 from transformers import DetrFeatureExtractor, DetrForSegmentation
@@ -61,7 +64,7 @@ def predict_animal_mask(im,
     #     color_mask[label_per_pixel==lbl,:] = color
 
     # Show image + mask
-    pred_img = np.array(image.convert('RGB'))*0.5 + color_mask*0.5
+    pred_img = np.array(image.convert('RGB'))*0.25 + color_mask*0.75
     pred_img = pred_img.astype(np.uint8)   
 
     return pred_img

requirements.txt

@@ -4,4 +4,5 @@ gradio
 numpy
 torch
 torchvision
-timm
+timm
+seaborn