subashdvorak commited on
Commit
c45d789
·
verified ·
1 Parent(s): 3af290e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def generate_graphs(new_story, metric):
31
  similarities = cosine_similarity([new_story_vector], knowledge_base_vectors)[0]
32
 
33
  # Filter indices with similarity > 0.85
34
- similar_indexes = np.where(similarities > 0.8)[0]
35
  print('Similar indexes:',similar_indexes)
36
 
37
  if len(similar_indexes) == 0:
 
31
  similarities = cosine_similarity([new_story_vector], knowledge_base_vectors)[0]
32
 
33
  # Filter indices with similarity > 0.85
34
+ similar_indexes = np.where(similarities > 0.7)[0]
35
  print('Similar indexes:',similar_indexes)
36
 
37
  if len(similar_indexes) == 0: