Update app.py

app.py

@@ -7,15 +7,14 @@ import subprocess
 import shlex
 import torch
 
-os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
-os.system("git clone https://github.com/Visual-AI/Mr.DETR.git MrDETR && cd MrDETR && rm -f requirements.txt && cd ..")
+os.system("git clone https://github.com/Visual-AI/vCLR.git && cd vCLR && rm -f requirements.txt && cd ..")
 subprocess.run(
     shlex.split(
         "pip install detrex-0.3.0-cp310-cp310-linux_x86_64.whl"
     )
 )
     
-sys.path.append("MrDETR/")
+sys.path.append("vCLR/")
 
 
 
@@ -28,7 +27,7 @@ from PIL import Image
 
 if __name__ == "__main__":
     gr.close_all()
-    cfg = LazyConfig.load("MrDETR/projects/mr_detr_align/configs/deformable_detr_swinl_two_stage_12ep_plusplus.py")
+    cfg = LazyConfig.load("vCLR/projects/vCLR_deformable_mask/configs/dino-resnet/deformable_train_voc_eval_nonvoc.py")
     cfg["model"].device = "cuda"
     cfg["train"].device = "cuda"
 
@@ -36,7 +35,7 @@ if __name__ == "__main__":
     # def 
     model = instantiate(cfg.model)
     checkpointer = DetectionCheckpointer(model)
-    checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")
+    checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_voc.pth")
 
     model.eval()
     model.cuda()
@@ -64,16 +63,9 @@ if __name__ == "__main__":
         ],
         outputs="image",
         examples=[
-            ["MrDETR/assets/000000014226.jpg", 0.5],
-            ["MrDETR/assets/000000028449.jpg", 0.3],
-            ["MrDETR/assets/000000070048.jpg", 0.5],
-            ["MrDETR/assets/000000218997.jpg", 0.5],
-            ["MrDETR/assets/000000279774.jpg", 0.5],
-            ["MrDETR/assets/000000434459.jpg", 0.5],
-            ["MrDETR/assets/000000448448.jpg", 0.5],
-            ["MrDETR/assets/000000560474.jpg", 0.5],
+            
         ],
-        title="[CVPR 2025 highlight] ***v***-CLR: View-Consistent Learning for Open-World Instance Segmentation",
+        title="[CVPR 2025 highlight] v-CLR: View-Consistent Learning for Open-World Instance Segmentation",
         description='''
             [![Paper](https://img.shields.io/badge/arXiv-2412.10028-red)](https://arxiv.org/abs/2412.10028)
             [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/mr-detr-instructive-multi-route-training-for/object-detection-on-coco-2017-val)](https://paperswithcode.com/sota/object-detection-on-coco-2017-val?p=mr-detr-instructive-multi-route-training-for)