ccclllwww commited on
Commit
7463f5d
·
verified ·
1 Parent(s): 1ccbf4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=1000
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