Update src/detect.py

src/detect.py

@@ -11,7 +11,7 @@ from src.utils import cells_to_bboxes, non_max_suppression, draw_predictions, Yo
 
 model = YOLOv3(num_classes=20)
 
-weights_path = "Final_trained_model.pth"
+weights_path = "/home/user/app/Final_trained_model.pth"
 ckpt = torch.load(weights_path, map_location="cpu")
 model.load_state_dict(ckpt)
 model.eval()