Gargi16 commited on
Commit
620cdca
·
verified ·
1 Parent(s): a44732b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -24,7 +24,8 @@ def search_courses(query):
24
  similarities = util.pytorch_cos_sim(query_embedding, course_embeddings)[0]
25
 
26
  # Find the top 5 most similar courses
27
- top_results = np.argsort(similarities, descending=True)[:5]
 
28
 
29
  # Prepare output
30
  results = []
 
24
  similarities = util.pytorch_cos_sim(query_embedding, course_embeddings)[0]
25
 
26
  # Find the top 5 most similar courses
27
+ #top_results = np.argsort(similarities, descending=True)[:5]
28
+ top_results = np.argsort(-similarities)[:5]
29
 
30
  # Prepare output
31
  results = []