tharu22 commited on
Commit
2fb20fd
Β·
1 Parent(s): 7f64752
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,7 @@ def get_image_embedding(image):
38
  return embedding
39
 
40
  # βœ… Function to Query Pinecone and Fetch Similar Images
41
- def search_similar_images(embedding, top_k=15):
42
  results = unsplash_index.query(
43
  vector=embedding,
44
  top_k=top_k,
@@ -60,7 +60,7 @@ if st.button("πŸ” Search by Text"):
60
  with st.spinner("Generating embedding..."):
61
  embedding = get_text_embedding(search_query)
62
  with st.spinner("Searching for similar images..."):
63
- matches = search_similar_images(embedding, top_k=5)
64
 
65
  st.subheader("πŸ”Ž Top Similar Images")
66
  for match in matches:
@@ -89,7 +89,7 @@ if uploaded_file:
89
  embedding = get_image_embedding(image)
90
 
91
  with st.spinner("Searching for similar images..."):
92
- matches = search_similar_images(embedding, top_k=5)
93
 
94
  st.subheader("πŸ”Ž Top Similar Images")
95
  for match in matches:
 
38
  return embedding
39
 
40
  # βœ… Function to Query Pinecone and Fetch Similar Images
41
+ def search_similar_images(embedding, top_k=10):
42
  results = unsplash_index.query(
43
  vector=embedding,
44
  top_k=top_k,
 
60
  with st.spinner("Generating embedding..."):
61
  embedding = get_text_embedding(search_query)
62
  with st.spinner("Searching for similar images..."):
63
+ matches = search_similar_images(embedding, top_k=10)
64
 
65
  st.subheader("πŸ”Ž Top Similar Images")
66
  for match in matches:
 
89
  embedding = get_image_embedding(image)
90
 
91
  with st.spinner("Searching for similar images..."):
92
+ matches = search_similar_images(embedding, top_k=10)
93
 
94
  st.subheader("πŸ”Ž Top Similar Images")
95
  for match in matches: