app.py

import streamlit as st 
from transformers import pipeline, set_seed
from datetime import datetime

def init():
    generator = setupAI()
    setupUI(generator)
    return generator

def setupAI():
    generator = pipeline('text-generation', model='gpt2')
    set_seed(datetime.now().second)
    return generator

def setupUI(generator):
    st.title('Phrase generator')
    phraseField = st.text_input('Enter a beginning phrase')
    if (st.button('Submit')):
        resultingphrase = phraseField.title()
        result = generator(phraseField.title(), max_length=30, num_return_sequences=5)
        generatedContent=''
        for item in result:
            generatedContent = generatedContent + item["generated_text"]

        st.success(generatedContent)

generator = init()