sess not defined error

[bjayakumar]

def _extend_conversation(conversation_as_string):
  generated_samples = gpt2.generate(
    sess,
    model_name=model_name,
    prefix=conversation_as_string,
    length=100,
    return_as_list = True
  )

I am getting error here if I run it on colab.

[bjayakumar]

This is the fix:

from nltk.tokenize import sent_tokenize
import gpt_2_simple as gpt2
sess = gpt2.start_tf_sess()
model_name = "774M"
gpt2.download_gpt2(model_name=model_name)
gpt2.load_gpt2(
    sess, 
  model_name=model_name
)

[ai-nikolai]

@bjayakumar thanks a lot for reporting the bug.

Feel free to open a PR

Otherwise, we will try and push a fix soon.