allencbzhang commited on
Commit
34cbd22
·
verified ·
1 Parent(s): 2ce01b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,7 +36,8 @@ if __name__ == "__main__":
36
  # def
37
  model = instantiate(cfg.model)
38
  checkpointer = DetectionCheckpointer(model)
39
- checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_voc.pth")
 
40
 
41
  model.eval()
42
  model.cuda()
 
36
  # def
37
  model = instantiate(cfg.model)
38
  checkpointer = DetectionCheckpointer(model)
39
+ # checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_voc.pth")
40
+ checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_coco.pth")
41
 
42
  model.eval()
43
  model.cuda()