LaotHF commited on
Commit
acf182b
·
verified ·
1 Parent(s): 64534ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,5 +10,5 @@ result = st.empty()
10
  llm = pipeline('text-generation', model='gpt2')
11
 
12
  if btn:
13
- res = llm.generation(text)
14
  result.success(res[0]["generated_text"].strip())
 
10
  llm = pipeline('text-generation', model='gpt2')
11
 
12
  if btn:
13
+ res = llm(text)
14
  result.success(res[0]["generated_text"].strip())