Revert "modify main.py"

This reverts commit ca5e73d3effdad95ce2e9e9417db585fb3bc3207.