Update app.py

app.py

@@ -20,7 +20,11 @@ from mmdet.utils import (compat_cfg, replace_cfg_vals, setup_multi_processes,
 
 import gradio as gr
 
-config_dict = OrderedDict([('swin-l-hdetr_sam-vit-b', 'projects/configs/hdetr/swin-l-hdetr_sam-vit-b.py'),
+config_dict = OrderedDict([('r50-hdetr_sam-vit-b', 'projects/configs/hdetr/r50-hdetr_sam-vit-b.py'),
+                           ('r50-hdetr_sam-vit-l', 'projects/configs/hdetr/r50-hdetr_sam-vit-l.py'),
+                           ('swin-t-hdetr_sam-vit-b', 'projects/configs/hdetr/swin-t-hdetr_sam-vit-b.py'),
+                           ('swin-t-hdetr_sam-vit-l', 'projects/configs/hdetr/swin-t-hdetr_sam-vit-l.py'),
+                           ('swin-l-hdetr_sam-vit-b', 'projects/configs/hdetr/swin-l-hdetr_sam-vit-b.py'),
                            ('swin-l-hdetr_sam-vit-l', 'projects/configs/hdetr/swin-l-hdetr_sam-vit-l.py'),
                            # ('swin-l-hdetr_sam-vit-h', 'projects/configs/hdetr/swin-l-hdetr_sam-vit-l.py'),
                            ('focalnet-l-dino_sam-vit-b', 'projects/configs/focalnet_dino/focalnet-l-dino_sam-vit-b.py'),
@@ -120,10 +124,10 @@ def main():
                         submit_btn = gr.Button(value="Submit")
                 output_img = gr.Image(type="numpy", label="Output")
             gr.Examples(
-                examples=[["./assets/img1.jpg", "swin-l-hdetr_sam-vit-b"],
-                          ["./assets/img2.jpg", "swin-l-hdetr_sam-vit-l"],
-                          ["./assets/img3.jpg", "swin-l-hdetr_sam-vit-l"],
-                          ["./assets/img4.jpg", "focalnet-l-dino_sam-vit-b"]],
+                examples=[["./assets/img1.jpg", "r50-hdetr_sam-vit-b"],
+                          ["./assets/img2.jpg", "r50-hdetr_sam-vit-b"],
+                          ["./assets/img3.jpg", "r50-hdetr_sam-vit-b"],
+                          ["./assets/img4.jpg", "r50-hdetr_sam-vit-b"]],
                 inputs=[input_img, model_type],
                 outputs=output_img,
                 fn=inference