Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,7 +113,7 @@ def generate_story_content(system_prompt: str, user_prompt: str) -> str:
|
|
| 113 |
# Generate text completion
|
| 114 |
generated_sequences = _model.generate(
|
| 115 |
**model_inputs,
|
| 116 |
-
max_new_tokens=
|
| 117 |
)
|
| 118 |
|
| 119 |
# Process and clean output
|
|
|
|
| 113 |
# Generate text completion
|
| 114 |
generated_sequences = _model.generate(
|
| 115 |
**model_inputs,
|
| 116 |
+
max_new_tokens=150
|
| 117 |
)
|
| 118 |
|
| 119 |
# Process and clean output
|