api key
Browse files
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=
|
| 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=
|
| 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=
|
| 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:
|