Update app.py

app.py

@@ -123,14 +123,14 @@ def main():
                     torch.cuda.empty_cache()
                     torch.manual_seed(0)
 
-                if all_output_texts:
-                    # Step 5: Generate story
-                    generate_story_button = st.button("Generate Story")
-
-                    if generate_story_button:
-                        story = generate_story(all_output_texts)
-                        st.write("Generated Story:")
-                        st.write(story)
+                    if all_output_texts:
+                        # Step 5: Generate story
+                        generate_story_button = st.button("Generate Story")
+    
+                        if generate_story_button:
+                            story = generate_story(all_output_texts)
+                            st.write("Generated Story:")
+                            st.write(story)
 
 if __name__ == "__main__":
     main()