allencbzhang commited on
Commit
687617e
·
verified ·
1 Parent(s): 6ab8d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -27,6 +27,7 @@ if __name__ == "__main__":
27
  checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")
28
 
29
  model.eval()
 
30
  vis_demo = VisualizationDemo(
31
  model=model,
32
  min_size_test=800,
 
27
  checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")
28
 
29
  model.eval()
30
+ model.cuda()
31
  vis_demo = VisualizationDemo(
32
  model=model,
33
  min_size_test=800,