miwojc commited on
Commit
49dcd1c
·
1 Parent(s): 985d236

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,6 +8,6 @@ st.text_input("Write something", value=st.session_state.user_text)
8
 
9
  if st.button("Write with transformer"):
10
  gpt2 = pipeline('text-generation')
11
- res = gpt2("My name is Mario and"])[0]["generated_text"]
12
  # st.session_state.user_text = res
13
  st.user_text = res
 
8
 
9
  if st.button("Write with transformer"):
10
  gpt2 = pipeline('text-generation')
11
+ res = gpt2("My name is Mario and")[0]["generated_text"]
12
  # st.session_state.user_text = res
13
  st.user_text = res