Update app.py

app.py

@@ -122,16 +122,16 @@ def main():
                     del image
                     torch.cuda.empty_cache()
                     torch.manual_seed(0)
-    st.write("Description of all images:")
-    st.write(all_output_texts)
-    if all_output_texts:
-        st.write("Generate story")
-        generate_story_button = st.button("Generate Story", key="generate_story")
-
-        if generate_story_button:
-            story = generate_story(all_output_texts)
-            st.write("Generated Story:")
-            st.write(story)
+        st.write("Description of all images:")
+        st.write(all_output_texts)
+        if all_output_texts:
+            st.write("Generate story")
+            generate_story_button = st.button("Generate Story", key="generate_story")
+    
+            if generate_story_button:
+                story = generate_story(all_output_texts)
+                st.write("Generated Story:")
+                st.write(story)
 
 if __name__ == "__main__":
     main()