Update generator.go

update import path of yaml