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).
Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment