Update app.py

app.py

@@ -2,11 +2,12 @@ from transformers import pipeline
 import streamlit as st
 
 # if 'user_text' not in st.session_state:
-st.session_state.user_text = 'foo'
+#    st.session_state.user_text = 'foo'
 
 st.text_input("Write something", value=st.session_state.user_text)
 
 if st.button("Write with transformer"):
     gpt2 = pipeline('text-generation')
-    res = gpt2(st.session_state["user_text"])[0]["generated_text"]
-    st.session_state.user_text = res
+    res = gpt2("My name is Mario and"])[0]["generated_text"]
+#    st.session_state.user_text = res
+    st.user_text = res