Update app.py

app.py

@@ -14,6 +14,8 @@ import numpy as np
 if __name__ == "__main__":
     gr.close_all()
     cfg = LazyConfig.load("MrDETR/projects/mr_detr_align/configs/deformable_detr_swinl_two_stage_12ep_plusplus.py")
+    cfg["model"].device = "cpu"
+    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")