Update Generator.py

Update