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 |
|