Michelangiolo commited on
Commit
a04b234
·
1 Parent(s): d4ac261
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = SentenceTransformer('all-mpnet-base-v2') #all-MiniLM-L6-v2 #all-mpnet-ba
15
  df = pd.read_parquet('df_encoded.parquet')
16
 
17
  #prepare model
18
- nbrs = NearestNeighbors(n_neighbors=8, algorithm='ball_tree').fit(df['text_vector_'].values.tolist())
19
 
20
  def search(df, query):
21
  product = model.encode(query).tolist()
 
15
  df = pd.read_parquet('df_encoded.parquet')
16
 
17
  #prepare model
18
+ nbrs = NearestNeighbors(n_neighbors=5, algorithm='ball_tree').fit(df['text_vector_'].values.tolist())
19
 
20
  def search(df, query):
21
  product = model.encode(query).tolist()