lalithadevi commited on
Commit
510185d
1 Parent(s): e48e7d5

Update find_similar_news.py

Browse files
Files changed (1) hide show
  1. 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)