Use for Custom Dataset

[MZeinab]

Thanks, it was so amazing.
I have one question. I'll be so grateful if you could help:

1. I watched the video that explained how to train yolov5 with the custom dataset. I used that notebook and exported the best.pt file, containing model weights. My question is: how to use it for inference?

• I used model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt') # PyTorch
  but It loads the original .pt file, not my custom one.