Update app.py

app.py

@@ -8,7 +8,6 @@ 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 PIL import Image
 import streamlit as st
 import zipfile
 
@@ -85,17 +84,13 @@ def plot_images(input_image_path: str, similar_img_paths: list):
     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)
+        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):
     search_params = {"metric_type": "L2"}