Update app.py

app.py

@@ -59,17 +59,21 @@ def main():
         top_p = st.slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95, step=0.05)
         repetition_penalty = st.slider("Repetition penalty", 1.0, 2.0, 1.0, step=0.05)
 
+    # Initialize session state for generated text
+    if "generated_text" not in st.session_state:
+        st.session_state.generated_text = ""
+
     # Generate button
     if st.button("Generate Waifu"):
         with st.spinner("Generating waifu character..."):
             prompt = format_prompt(name, hair_color, personality, outfit_style, hobbies, favorite_food, background_story, system_prompt)
-            generated_text = generate_text(prompt, temperature, max_new_tokens, top_p, repetition_penalty)
+            st.session_state.generated_text = generate_text(prompt, temperature, max_new_tokens, top_p, repetition_penalty)
             st.success("Waifu character generated!")
 
     # Display the generated character
-    if generated_text:
+    if st.session_state.generated_text:
         st.subheader("Generated Waifu Character")
-        st.write(generated_text)
+        st.write(st.session_state.generated_text)
 
 if __name__ == "__main__":
     main()