Spaces:
Runtime error
Runtime error
Commit ·
a04b234
1
Parent(s): d4ac261
changes
Browse files
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=
|
| 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()
|