Update helper_functions.py

helper_functions.py

@@ -31,6 +31,7 @@ def search(df, query, max_n, max_token):
     )
     df["similarity"] = df.ada_embedding.apply(lambda x: cosine_similarity(x, query_embedding))
     df = df.sort_values("similarity", ascending=False).head(max_n)
+    df = df[df['similarity'] >= .77]
     df["cumulative_sum"] = df.token.cumsum()
     return '\n\n'.join(df[(df['cumulative_sum'] < max_token)]["combined_summarised"])