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(
|