Article generation in website

article_generation.py

#Import streamlit library
import streamlit as st

st.title("Medium Article Generator") #set title of website
starting_sentence = st.text_input(label='Enter Starting Sentence') #set input and label

#Function to run model and generate new text
@st.cache
def generate():
    import gpt_2_simple as gpt2

    global sess1
    sess = gpt2.start_tf_sess()
    gpt2.load_gpt2(sess)

    global single_text
    single_text = gpt2.generate(sess, prefix=starting_sentence, top_k=40, length=100,
                                return_as_list=True)[0]
    return single_text


model_load_state = st.text("Loading Gpt2 model ... ")
generate() #Run function after input
model_load_state.text('Loading Gpt2 ... Done! Have Fun:)')

st.write(single_text) #write output generated by model