Create app.py

app.py

@@ -0,0 +1,12 @@
+from transformers import pipeline
+import streamlit as st
+
+if 'user_text' not in st.session_state:
+    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