Update app.py

app.py

@@ -32,7 +32,7 @@ def clean_generated_text(text):
     clean_text = re.sub(r'</s>$', '', text).strip()
     return clean_text
 
-def generate_text(prompt, temperature=0.9, max_new_tokens=1512, top_p=0.95, repetition_penalty=1.0):
+def generate_text(prompt, temperature=0.9, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
     temperature = max(temperature, 1e-2)
     generate_kwargs = dict(
         temperature=temperature,
@@ -160,7 +160,8 @@ def main():
                     )
 
                     # Generate character description
-                    st.session_state.character_description = generate_text(description_prompt, temperature, max_new_tokens, top_p, repetition_penalty)
+                    character_description = generate_text(description_prompt, temperature, max_new_tokens, top_p, repetition_penalty)
+                    st.session_state.character_description = character_description
                     
                     # Generate image prompt
                     st.session_state.image_prompt = generate_text(image_prompt, temperature, max_new_tokens, top_p, repetition_penalty)