Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|