Update app.py

app.py

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