Update app.py

app.py

@@ -16,9 +16,13 @@ st.write(f"{x} squared is {x * x}")
 uploaded_images = st.file_uploader("Upload images", type=["jpg", "png"], accept_multiple_files=True)
 
 # Display thumbnail images alongside file names and sizes in the sidebar
+@st.cache
+def load_image(img):
+    return Image.open(img)
+
 if uploaded_images:
     for idx, img in enumerate(uploaded_images):
-        image = Image.open(img)
+        image = load_image(img)
         # Generate unique keys for image display and checkbox
         image_key = f"{img.name}_image_{idx}"
         checkbox_key = f"{img.name}_checkbox_{idx}"
@@ -32,7 +36,7 @@ for idx, img in enumerate(uploaded_images):
     checkbox_key = f"{img.name}_checkbox_{idx}"
     selected = st.sidebar.checkbox(f"Select {img.name}", value=False, key=checkbox_key)
     if selected:
-        selected_images.append(Image.open(img))
+        selected_images.append(load_image(img))
 
 if st.button("Predict Emotions") and selected_images:
     if len(selected_images) == 2: