diff --git a/docker_test.py b/docker_test.py index 8ecfaad..b561cd0 100644 --- a/docker_test.py +++ b/docker_test.py @@ -2,7 +2,7 @@ import torch import cv2 -model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.pyy", "weights/groundingdino_swint_ogc.pth") +model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.py", "weights/groundingdino_swint_ogc.pth") model = model.to('cuda:0') print(torch.cuda.is_available()) -print('DONE!') \ No newline at end of file +print('DONE!')