Committing changes to main.py that works locally.

first cut to make sure I can run in Spaces