UI doesn't show up in colab when using the faces dataset

[aferriss]

Hi! Thanks for making this code available. I saw your lecture in Ali Jahanian's class on yt and wanted to try this out with faces.

However, when changing the colab to use faces instead of church, the UI doesn't show up when the cell completes. There's no error or anything as far as I can tell, execution just ends without anything happening. Things work correctly when I leave the modelname as church though.

Thanks again for the help!

[dvschultz]

seems like a colab issue. Im able to run it locally in jupyter but not on Colab. See this issue: https://github.com/davidbau/rewriting/issues/6