Update src/detect.py

src/detect.py

@@ -12,7 +12,7 @@ from src.utils import cells_to_bboxes, non_max_suppression, draw_predictions, Yo
 
 
 weights_path = "/home/user/app/model_ass_13_up.ckpt"
-model = Assignment13().load_from_checkpoint(weights_path,map_location=torch.device("cpu"))
+model = Assignment13.load_from_checkpoint(weights_path,map_location=torch.device("cpu"))
 model = model.model
 #ckpt = torch.load(weights_path, map_location="cpu")
 #model.load_state_dict(ckpt)