Update app.py

app.py

@@ -60,7 +60,7 @@ def reply(input, dataset_name):
         input = input
         input_vector = generate_embeddings(input, model_id,hf_token)
         df["similarities"]=df["embedding"].apply(lambda x: cosine_similarity([x],[input_vector])[0][0])
-        data = df.sort_values("similarities", ascending=False).head(10)
+        data = df.sort_values("similarities", ascending=False).head(5)
         data.to_csv("sorted.csv")
         context = []
         for i, row in data.iterrows():