Update app.py

app.py

@@ -118,20 +118,16 @@ def main():
                         st.write(description)
                         all_output_texts.append(description)
 
-                    # Clear memory after processing each file
-                    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)
+                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()