;

app.py

@@ -4,13 +4,21 @@ from datetime import datetime
 
 def init():
     generator = setupAI()
+    setupUI()
     return generator
 
 def setupAI():
     generator = pipeline('text-generation', model='gpt2')
     set_seed(datetime.now().second)
     return generator
-    
+
+def setupUI():
+    st.title('Phrase generator')
+    phrase = st.text_input('Enter a beginning phrase')
+    if (st.button('Submit')):
+        resultingphrase = phrase.title()
+        st.success(resultingphrase)
+
 generator = init()
 result = generator("I come in peace.", max_length=30, num_return_sequences=5)
 for item in result: