Update run.py and sync changes to HF

Cleanup: Remove debug prints, improve logging, and refactor comments