Update app.py

app.py

@@ -15,8 +15,8 @@ from PIL import Image
 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"
+    cfg["model"].device = "cuda"
+    cfg["train"].device = "cuda"
     model = instantiate(cfg.model)
     checkpointer = DetectionCheckpointer(model)
     checkpointer.load("https://github.com/Visual-AI/Mr.DETR/releases/download/weights/MrDETR_align_swinL_12ep_900q_safe.pth")