Update app.py

app.py

@@ -105,8 +105,11 @@ def main():
         # Convert the uploaded image to an array using the read_image function
         image = read_image(uploaded_image)
 
-        # Display the uploaded image
-        st.image(image, caption="Query Image", width=224, height = 224)
+        image = Image.fromarray(image)
+        image = image.resize((224, 224))
+
+        # Display the resized image
+        st.image(image, caption="Query Image", width=224)
 
         # Find similar images
         similar_images, length = find_similar_images(image)
@@ -114,6 +117,8 @@ def main():
         # Calculate the total number of pages required
         total_pages = (length + images_per_page - 1) // images_per_page
 
+        page_number = st.slider("Select Page", min_value=1, max_value=num_pages)
+
         # Calculate the range of images to display for the selected page
         start_index = (page_number - 1) * images_per_page
         end_index = min(start_index + images_per_page, length)
@@ -145,12 +150,6 @@ def main():
                 st.write(f"Could not find details for image with ID: {img_id}")
         
 
-    with st.container():
-        st.write("Page Selector")
-        page_number = st.number_input("Page", min_value=1, max_value=20, step=1, value=1)
-
-    
-        
 
 if __name__ == '__main__':
     main()