Can sample-beam.lua be modified to process at word level?

[ghost]

I've been experimenting your word version and I was wondering if it's possible to modify 'sample-beam.lua' to work properly at the word level?

https://github.com/pender/char-rnn/blob/master/sample-beam.lua

It improves the output results at the character level but current sampling output is concatenated at word level:

Thequickbrownfoxjumpedoverthelazydog.

Instead of

The quick brown fox jumped over the lazy dog.

I'm new to lua/torch and would appreciate any suggestions to modify the code.

Cheers,