Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -119,7 +119,7 @@ class Chatbot():
|
|
| 119 |
embeddings = np.vstack(df.text.apply(lambda x: get_embedding(x, engine=embedding_model)))
|
| 120 |
return embeddings
|
| 121 |
|
| 122 |
-
def search_embeddings(self, df, query, n=3,
|
| 123 |
|
| 124 |
# Step 1. Get an embedding for the question being asked to the PDF
|
| 125 |
query_embedding = get_embedding(
|
|
|
|
| 119 |
embeddings = np.vstack(df.text.apply(lambda x: get_embedding(x, engine=embedding_model)))
|
| 120 |
return embeddings
|
| 121 |
|
| 122 |
+
def search_embeddings(self, embeddings, df, query, n=3, pprint=True):
|
| 123 |
|
| 124 |
# Step 1. Get an embedding for the question being asked to the PDF
|
| 125 |
query_embedding = get_embedding(
|