Create thop_main.py