Tesneem commited on
Commit
eb153cc
·
verified ·
1 Parent(s): 3d14986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = adjusted_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
 
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