init

app.py

@@ -1,4 +1,5 @@
 import os
+from pydoc import describe
 os.system('pip install torch==1.9 torchvision')
 os.system('pip install detectron2==0.5 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
 os.system('pip install timm opencv-python-headless')
@@ -14,7 +15,7 @@ from opendet2 import add_opendet_config
 model_cfgs = {
     "FR-CNN": ["configs/faster_rcnn_R_50_FPN_3x_baseline.yaml", "frcnn_r50.pth"],
     "OpenDet-R50": ["configs/faster_rcnn_R_50_FPN_3x_opendet.yaml", "opendet2_r50.pth"],
-    "OpenDet-SwinT": ["configs\faster_rcnn_Swin_T_FPN_3x_opendet.yaml", "opendet2_swint.pth"],
+    "OpenDet-SwinT": ["configs/faster_rcnn_Swin_T_FPN_3x_opendet.yaml", "opendet2_swint.pth"],
 }
 
 
@@ -39,6 +40,7 @@ def inference(input, model):
     output = visualized_output.get_image()[:, :, ::-1]
     return output
 
+examples = [[x, "OpenDet-R50"] for x in os.listdir('demo') if x.endswith(".jpg")]
 
 iface = gr.Interface(
     inference,
@@ -47,6 +49,11 @@ iface = gr.Interface(
         gr.inputs.Radio(
             ["FR-CNN", "OpenDet-R50", "OpenDet-SwinT"], default='OpenDet-R50'),
     ],
-    "image")
-
-iface.launch()
+    "image",
+    examples=examples,
+    title="OpenDet",
+    article="<p style='text-align: center'><a href=https://github.com/csuhan/opendet2' target='_blank'>Github Repo</a> | <a href=https://csuhan.com' target='_blank'>Author Page</a></p>",
+    description="Online demo for: Expanding Low-Density Latent Regions for Open-Set Object Detection. Please upload your image, or click one of the examples to load them",
+)
+
+iface.launch(enable_queue=True, cache_examples=True)