Update app.py

app.py

@@ -8,7 +8,7 @@ from tensorflow.keras.applications.vgg19 import preprocess_input
 from tensorflow.keras.layers import GlobalAveragePooling2D
 from pymilvus import connections, Collection, utility
 from requests import get
-from shutil import rmtree
+from PIL import Image
 import streamlit as st
 import zipfile
 
@@ -90,7 +90,11 @@ def plot_images(input_image_path: str, similar_img_paths: list):
     
         # display similar images
         st.markdown('<p style="font-size: 20px; font-weight: bold">Similar images</p>', unsafe_allow_html=True)
-        st.pyplot(fig)
+        # st.pyplot(fig)
+        
+        fig.savefig("similar_images.png", bbox_inches='tight', dpi=100)
+        image = Image.open('similar_images.png')
+        st.image(image, output_format="PNG")
 
 
 def find_similar_images(img_path: str, top_n: int=15):