Update app.py
Browse files
app.py
CHANGED
@@ -33,13 +33,13 @@ def vector_search(query):
|
|
33 |
# Calculate cosine similarity between the query and all embeddings
|
34 |
similarities = cosine_similarity([query_embedding], embeddings)
|
35 |
|
36 |
-
# Adjust similarity scores based on ratings
|
37 |
-
ratings = np.array([item["rating"] for item in data_list])
|
38 |
-
adjusted_similarities = similarities * ratings.reshape(-1, 1)
|
39 |
|
40 |
# Get top N most similar items (e.g., top 3)
|
41 |
top_n = 3
|
42 |
-
top_indices =
|
43 |
top_items = [data_list[i] for i in top_indices]
|
44 |
|
45 |
# Format the output for display
|
|
|
33 |
# Calculate cosine similarity between the query and all embeddings
|
34 |
similarities = cosine_similarity([query_embedding], embeddings)
|
35 |
|
36 |
+
# # Adjust similarity scores based on ratings
|
37 |
+
# ratings = np.array([item["rating"] for item in data_list])
|
38 |
+
# adjusted_similarities = similarities * ratings.reshape(-1, 1)
|
39 |
|
40 |
# Get top N most similar items (e.g., top 3)
|
41 |
top_n = 3
|
42 |
+
top_indices = similarities[0].argsort()[-top_n:][::-1] # Get indices of the top N results
|
43 |
top_items = [data_list[i] for i in top_indices]
|
44 |
|
45 |
# Format the output for display
|