Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"
|
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
|