saritha5 commited on
Commit
f637ac1
·
1 Parent(s): d600cc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -35,3 +35,8 @@ datafile_path = "fine_food_reviews_with_embeddings_1k.csv"
35
  df = pd.read_csv(datafile_path)
36
  df["embedding"] = df.embedding.apply(eval).apply(np.array)
37
 
 
 
 
 
 
 
35
  df = pd.read_csv(datafile_path)
36
  df["embedding"] = df.embedding.apply(eval).apply(np.array)
37
 
38
+ prompt = input("What do you want to search for? : ")
39
+ top_n = int(input("How many results do you want to see? : "))
40
+ print()
41
+ results,product = search_reviews(df, prompt, top_n)
42
+