Spaces:

Michelangiolo
/

startup-finder

Runtime error

Michelangiolo commited on Mar 31, 2023

Commit

942976d

•

1 Parent(s): 6d7d107

changes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ import pandas as pd
 from sentence_transformers import SentenceTransformer
 #prepare model
-nbrs = NearestNeighbors(n_neighbors=5, algorithm='ball_tree').fit(df_knn)
 def search(query):
     product = model.encode(query).tolist()

 from sentence_transformers import SentenceTransformer
 #prepare model
+nbrs = NearestNeighbors(n_neighbors=20, algorithm='ball_tree').fit(df_knn)
 def search(query):
     product = model.encode(query).tolist()