lalithadevi
commited on
Commit
•
510185d
1
Parent(s):
e48e7d5
Update find_similar_news.py
Browse files- find_similar_news.py +1 -1
find_similar_news.py
CHANGED
@@ -43,7 +43,7 @@ def get_milvus_collection():
|
|
43 |
logger.warning('Exiting get_milvus_collection()')
|
44 |
return collection
|
45 |
|
46 |
-
def find_similar_news(search_vec, collection, vectorizer, sent_model, ce_model, top_n: int=5):
|
47 |
logger.warning('Entering find_similar_news')
|
48 |
search_params = {"metric_type": "IP"}
|
49 |
# search_vec = vectorizer.vectorize_(text)
|
|
|
43 |
logger.warning('Exiting get_milvus_collection()')
|
44 |
return collection
|
45 |
|
46 |
+
def find_similar_news(text, search_vec, collection, vectorizer, sent_model, ce_model, top_n: int=5):
|
47 |
logger.warning('Entering find_similar_news')
|
48 |
search_params = {"metric_type": "IP"}
|
49 |
# search_vec = vectorizer.vectorize_(text)
|