``` 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.