Update pages/📸_InstaMuse.py

pages/📸_InstaMuse.py

@@ -54,6 +54,9 @@ with st.sidebar:
 # Main page content
 st.markdown('<p class="big-font">InstaMuse Photo Caption Generator</p>', unsafe_allow_html=True)
 st.write("### Upload your photo below and spark some caption magic! ଘ(੭ˊᵕˋ)੭* ੈ✩‧₊")
+st.write(
+'**Notice**: *The first time you upload an image, it may take longer to generate captions.*\n*Subsequent uploads will be faster.*')
+
 
 # Upload image file and process image
 file = st.file_uploader(
@@ -66,16 +69,6 @@ if file:
     image = Image.open(file)
     image.thumbnail((600, 600), Image.Resampling.LANCZOS)
 
-    try:
-        desc = caption_generator.image_2_text(image)
-        captions, caption_list, img_description = caption_generator.text_2_caption(desc)
-        st.session_state['captions'] = captions
-        st.session_state['caption_list'] = caption_list
-        st.session_state['img_description'] = img_description
-    except Exception as e:
-        st.error(f"Error generating captions: {e}")
-        captions = None
-
     with st.spinner(r'#### :sparkles: :sparkles: Generating... please wait :hourglass_flowing_sand:'):
         st.session_state['file'] = file
 
@@ -85,6 +78,15 @@ if file:
             with col1:
                 st.markdown("## 📸 Your Image:")
                 st.image(image, caption='Uploaded Image', use_column_width=True)
+                    try:
+                        desc = caption_generator.image_2_text(image)
+                        captions, caption_list, img_description = caption_generator.text_2_caption(desc)
+                        st.session_state['captions'] = captions
+                        st.session_state['caption_list'] = caption_list
+                        st.session_state['img_description'] = img_description
+                    except Exception as e:
+                        st.error(f"Error generating captions: {e}")
+                        captions = None
 
             with col2:
                 if captions: