Edited docker_test.py file

2024-05-03 21:40:53 -04:00 · 2024-05-03 21:40:53 -04:00 · f321f27a6f
parent 16e0ccdb7d
commit f321f27a6f
1 changed files with 1 additions and 1 deletions
--- a/docker_test.py
+++ b/docker_test.py
@ -2,7 +2,7 @@ from groundingdino.util.inference import load_model, load_image, predict, annota
 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!')