Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Select an option

  • Save saimadhu-polamuri/ccc5eed727741d52b371f7d61a27f948 to your computer and use it in GitHub Desktop.

Select an option

Save saimadhu-polamuri/ccc5eed727741d52b371f7d61a27f948 to your computer and use it in GitHub Desktop.
def generate_text(seed_text, num_words):
for _ in range(num_words):
token_list = tokenizer.texts_to_sequences([seed_text])[0]
token_list = tf.keras.preprocessing.sequence.pad_sequences([token_list], maxlen=max_length)
predicted_index = model.predict(token_list)[0].argmax()
predicted_word = tokenizer.index_word[predicted_index]
seed_text += " " + predicted_word
return seed_text
# Generate a story
seed_text = "Once upon a time"
generated_text = generate_text(seed_text, 100)
print(generated_text)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment