Update app.py

app.py

@@ -26,9 +26,13 @@ models = [
         "name": "Instance Segmentation",
         "config_file": "COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml",
     },
+    {
+        "name": "Panoptic Segmentation",
+        "config_file": "COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml",
+    },
     {
         "name": "Custom Model",
-        "config_file": "COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml",  
+        "config_file": "COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml",  
         "model_path": "https://huggingface.co/spaces/eyepop-ai/segmentation/blob/main/model_final.pth",  
     },
 ]
@@ -80,9 +84,10 @@ def inference(image_url, image, min_score, model_name):
 title = "# Segmentation Model Demo"
 description = """
 This demo introduces an interactive playground for pretrained Detectron2 model.
-Currently, two models are supported that were trained on COCO dataset:
+Currently, two models are supported that were trained on COCO and custom datasets:
 * [Instance Segmentation](https://github.com/facebookresearch/detectron2/blob/main/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml): Identifies, outlines individual object instances.
 * [Panoptic Segmentation](https://github.com/facebookresearch/detectron2/blob/main/configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x.yaml): Unifies instance and semantic segmentation.
+* [Custom Model](https://huggingface.co/spaces/eyepop-ai/segmentation/blob/main/model_final.pth): Identifies, outlines rounded objects in petri dishes.
 """
 footer = "Made by eyepop.ai with ❤️."