Spaces:

allencbzhang
/

Mr.DETR

Running on Zero

allencbzhang commited on Apr 2

Commit

d044d4b

verified ·

1 Parent(s): b2605e8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     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(