I've encountered a size mismatch issue when trying to run the author's pretrained model on a single image. Could someone who have successfully run the model share your environment.yml file and the running command? Lot of thanks! (The error screenshot is as follows.)
