Adr740 commited on
Commit
a3021eb
·
1 Parent(s): 66ae10b

Update get_hadiths.py

Browse files
Files changed (1) hide show
  1. get_hadiths.py +1 -1
get_hadiths.py CHANGED
@@ -50,7 +50,7 @@ class HadithSearch:
50
  embedding = self._get_embedding(user_input, model='text-embedding-ada-002')
51
  self.data['similarities'] = self.data.embeding.apply(lambda x: self._cosine_similarity(x, embedding))
52
 
53
- results = self.data.sort_values('similarities', ascending=False).head(int(num_hadiths)).to_dict(orient="records")
54
  formatted_results = self._format_results(results)
55
  return formatted_results
56
 
 
50
  embedding = self._get_embedding(user_input, model='text-embedding-ada-002')
51
  self.data['similarities'] = self.data.embeding.apply(lambda x: self._cosine_similarity(x, embedding))
52
 
53
+ results = self.data.sort_values('similarities', ascending=False).head(int(num_hadiths))
54
  formatted_results = self._format_results(results)
55
  return formatted_results
56