Spaces:

allencbzhang
/

Mr.DETR

Running on Zero

allencbzhang commited on Apr 2

Commit

8d27209

verified ·

1 Parent(s): d044d4b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,8 +18,7 @@ if __name__ == "__main__":
     cfg["train"].device = "cpu"
     model = instantiate(cfg.model)
     checkpointer = DetectionCheckpointer(model)
-    os.system("wget https://huggingface.co/allencbzhang/Mr.DETR/blob/main/MrDETR_align_swinL_12ep_900q.pth")
-    checkpointer.load("MrDETR_align_swinL_12ep_900q.pth")
     model.eval()
     vis_demo = VisualizationDemo(

     cfg["train"].device = "cpu"
     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")
     model.eval()
     vis_demo = VisualizationDemo(