Spaces:
Sleeping
Sleeping
Update get_hadiths.py
Browse files- 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))
|
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 |
|