Spaces:
Sleeping
Sleeping
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
|