app.py

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

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

def setupAI():
    generator = pipeline('text-generation', model='gpt2')
    set_seed(datetime.now().second)
    return generator
    
generator = init()
result = generator("I come in peace.", max_length=30, num_return_sequences=5)
for item in result:
    st.write (item["generated_text"])