vjain commited on
Commit
2f5a6ea
·
1 Parent(s): bb67458

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def reply(input, dataset_name):
45
  return "Please Enter a Question to get an Answer"
46
  df = dataframes[dataset_name]
47
  input = input
48
- input_vector = generate_embeddings(input, model_id,hf_token)
49
  df["similarities"]=df["embedding"].apply(lambda x: cosine_similarity([x],[input_vector])[0][0])
50
  data = df.sort_values("similarities", ascending=False).head(10)
51
  data.to_csv("sorted.csv")
 
45
  return "Please Enter a Question to get an Answer"
46
  df = dataframes[dataset_name]
47
  input = input
48
+ input_vector = generate_embeddings(input, model_id,hf_token, dataset_name)
49
  df["similarities"]=df["embedding"].apply(lambda x: cosine_similarity([x],[input_vector])[0][0])
50
  data = df.sort_values("similarities", ascending=False).head(10)
51
  data.to_csv("sorted.csv")