Cachoups commited on
Commit
570b200
·
verified ·
1 Parent(s): 1aa667b

Update lib/comparison.py

Browse files
Files changed (1) hide show
  1. lib/comparison.py +2 -2
lib/comparison.py CHANGED
@@ -42,7 +42,7 @@ def compare_summaries(paragraph, paragraphs):
42
 
43
  # Find the most similar summary
44
  most_similar_index = np.argmax(similarities) # Get index of most similar summary
45
- most_similar_summary = summaries[most_similar_index] # Corresponding summary
46
  similarity_score = similarities[most_similar_index] # Similarity score
47
 
48
- return most_similar_summary, similarity_score
 
42
 
43
  # Find the most similar summary
44
  most_similar_index = np.argmax(similarities) # Get index of most similar summary
45
+ most_similar_summary = paragraphs[most_similar_index] # Corresponding summary
46
  similarity_score = similarities[most_similar_index] # Similarity score
47
 
48
+ return most_similar_summary