How can I use the demo (inference) on cpu?

[exnx]

Hi, is there a way I can use the demo.py on cpu? When I run it and it tries to load, it breaks because the pretrained model is for gpu. I couldn't find the option / flag to turn off cuda. Is that possible for the demo script? Thanks!

Eric