;

app.py

@@ -4,7 +4,7 @@ from datetime import datetime
 
 def init():
     generator = setupAI()
-    setupUI()
+    setupUI(generator)
     return generator
 
 def setupAI():
@@ -12,14 +12,16 @@ def setupAI():
     set_seed(datetime.now().second)
     return generator
 
-def setupUI():
+def setupUI(generator):
     st.title('Phrase generator')
-    phrase = st.text_input('Enter a beginning phrase')
+    phraseField = st.text_input('Enter a beginning phrase')
     if (st.button('Submit')):
-        resultingphrase = phrase.title()
-        st.success(resultingphrase)
+        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()
-result = generator("I come in peace.", max_length=30, num_return_sequences=5)
-for item in result:
-    st.write (item["generated_text"])