I have weights in .pth file and confgiguarion file in YAML format, i want to perform prediction (Detection of Building) on test images and utilizing the following code for t