Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ text = st.text_input('Please input')
|
|
| 7 |
btn = st.button('Send')
|
| 8 |
result = st.empty()
|
| 9 |
|
| 10 |
-
llm =
|
| 11 |
|
| 12 |
if btn:
|
| 13 |
res = llm.generation(text)
|
|
|
|
| 7 |
btn = st.button('Send')
|
| 8 |
result = st.empty()
|
| 9 |
|
| 10 |
+
llm = pipeline('text-generation', model='gpt2')
|
| 11 |
|
| 12 |
if btn:
|
| 13 |
res = llm.generation(text)
|