Remove useless import in generate.py