added missing imports
#12
by
bitsTobyte
- opened
- import statements were missing
- max_length parameter in generate function was set to lower value than the input tokens, which was producing error, set it to little high than input tokens (from 10 to 64 as 48 are input tokens).