Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 = "
|
19 |
-
cfg["train"].device = "
|
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")
|