qgallouedec HF staff commited on
Commit
729f92f
β€’
1 Parent(s): 4afacff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def generate_text(model_name, input_text):
32
  yield "Generating text..."
33
 
34
  # Generate text
35
- generated_text = generator(input_text)[0]["generated_text"]
36
 
37
  # Return the generated text
38
  yield generated_text
 
32
  yield "Generating text..."
33
 
34
  # Generate text
35
+ generated_text = generator(input_text, max_length=100)[0]["generated_text"]
36
 
37
  # Return the generated text
38
  yield generated_text