Thank you for the response! That is good insight, but I was also trying to use the default model configuration for MobileNetV2 model and had no issues in the simulator. The issue was only with the model I trained. My model would work using the arm64 simulator or when I installed on my physical iPhone, just not in the iOS simulator.

This in addition to taking out my geometry reader worked. Thanks.