Update app.py

app.py

@@ -16,6 +16,7 @@ import zipfile
 with zipfile.ZipFile("Vegetable Images.zip", 'r') as zip_ref:
     zip_ref.extractall('.')
 
+placeholder = st.empty()
 
 class ImageVectorizer:
     '''
@@ -73,14 +74,15 @@ def plot_images(input_image_path: str, similar_img_paths: list):
     # display input image
     rcParams.update({'figure.autolayout': True})
     input_image = load_img(input_image_path, color_mode="rgb", target_size=(224, 224))
-    st.markdown('<p style="font-size: 20px; font-weight: bold">Input image</p>', unsafe_allow_html=True)
-    st.image(input_image)
-
-    st.write('  \n')
-
-    # display similar images
-    st.markdown('<p style="font-size: 20px; font-weight: bold">Similar images</p>', unsafe_allow_html=True)
-    st.pyplot(fig)
+    with placeholder.container():
+        st.markdown('<p style="font-size: 20px; font-weight: bold">Input image</p>', unsafe_allow_html=True)
+        st.image(input_image)
+    
+        st.write('  \n')
+    
+        # display similar images
+        st.markdown('<p style="font-size: 20px; font-weight: bold">Similar images</p>', unsafe_allow_html=True)
+        st.pyplot(fig)
 
 
 def find_similar_images(img_path: str, top_n: int=15):
@@ -131,7 +133,7 @@ def main():
         st.markdown(desc, unsafe_allow_html=True)
         img_url = st.text_input("Paste the image URL of a vegetable and hit Enter:", "")
         if img_url:
-            st.empty()
+            placeholder.empty()
             img_path = process_input_image(img_url)
             find_similar_images(img_path, 15)
             delete_file(os.path.dirname(img_path))