generate_app.py

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 
from tensorflow.keras.preprocessing.sequence import pad_sequences

def poem_generator(input_sentence,model,tokenizer,input_length):
	user_input = input_sentence
	output = ""
	in_text = user_input.lower()
	start = ' '+ in_text+'\n'
	output += start
	for i in range(99):
		encoded = tokenizer.texts_to_sequences([in_text])[0]
		encoded = pad_sequences([encoded], maxlen = input_length, truncating = 'pre')
		yhat = model.predict_classes(encoded, verbose = 0)
		out_word = ''
		for word, index in tokenizer.word_index.items():
			if index == yhat:
				out_word = word
				break
		in_text += ' ' + out_word
		out_word = ' ' + out_word
		if i % 7 ==0 and i !=0:
			out_word = out_word + '\n'
		output += out_word

	return output