merve HF staff commited on
Commit
07bb839
β€’
1 Parent(s): d4e9ffa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import pipeline, set_seed
7
  def infer(sent, max_length, num_return_sequences):
8
  generator = pipeline('text-generation', model='gpt2')
9
 
10
- return generator(sent, max_length, num_return_sequences)
11
 
12
 
13
  st.title("Write with Transformers πŸ¦„")
 
7
  def infer(sent, max_length, num_return_sequences):
8
  generator = pipeline('text-generation', model='gpt2')
9
 
10
+ return generator(sent, max_length=max_length, num_return_sequences=num_return_sequences)
11
 
12
 
13
  st.title("Write with Transformers πŸ¦„")