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