Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,7 +6,7 @@ import os
 
 
 from models.imageCaptioning import generateCaption
-from models.storyGeneration import generateStory
+# from models.storyGeneration import generateStory
 
 st.set_page_config(page_title="Image2Story Generator", layout="centered")
 st.title("Image2Story Generator")
@@ -34,7 +34,7 @@ if uploaded_file:
         captions = generateCaption(image_path)
         st.success(f"The image is of : {captions}")
     
-    with st.spinner("Generating Story..."): 
-        story = generateStory(captions)
-        st.markdown("Generated Story")
-        st.write(story)
+    # with st.spinner("Generating Story..."): 
+    #     story = generateStory(captions)
+    #     st.markdown("Generated Story")
+    #     st.write(story)