From f321f27a6f7759e298134e05323985fac23bd5bf Mon Sep 17 00:00:00 2001 From: jvarshini99 Date: Fri, 3 May 2024 21:40:53 -0400 Subject: [PATCH] Edited docker_test.py file --- docker_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker_test.py b/docker_test.py index 8ecfaad..78700ba 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