allencbzhang commited on
Commit
02a0351
·
verified ·
1 Parent(s): 9835bb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ from PIL import Image
15
  if __name__ == "__main__":
16
  gr.close_all()
17
  cfg = LazyConfig.load("MrDETR/projects/mr_detr_align/configs/deformable_detr_swinl_two_stage_12ep_plusplus.py")
18
- cfg["model"].device = "cpu"
19
- cfg["train"].device = "cpu"
20
  model = instantiate(cfg.model)
21
  checkpointer = DetectionCheckpointer(model)
22
  checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")
 
15
  if __name__ == "__main__":
16
  gr.close_all()
17
  cfg = LazyConfig.load("MrDETR/projects/mr_detr_align/configs/deformable_detr_swinl_two_stage_12ep_plusplus.py")
18
+ cfg["model"].device = "cuda"
19
+ cfg["train"].device = "cuda"
20
  model = instantiate(cfg.model)
21
  checkpointer = DetectionCheckpointer(model)
22
  checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")