Spaces:
Runtime error
Runtime error
bugfix: Fix embed_queries() KeyError
#1
by
kmclester
- opened
app.py
CHANGED
@@ -21,7 +21,7 @@ def start_haystack():
|
|
21 |
def load_data_to_store(document_store, retriever):
|
22 |
df = pd.read_csv('monopoly_qa-v1.csv')
|
23 |
questions = list(df.Question)
|
24 |
-
df['embedding'] = retriever.embed_queries(
|
25 |
df = df.rename(columns={"Question":"content","Answer":"answer"})
|
26 |
df.drop('link to source (to prevent duplicate sources)',axis=1, inplace=True)
|
27 |
|
|
|
21 |
def load_data_to_store(document_store, retriever):
|
22 |
df = pd.read_csv('monopoly_qa-v1.csv')
|
23 |
questions = list(df.Question)
|
24 |
+
df['embedding'] = retriever.embed_queries(queries=questions).tolist()
|
25 |
df = df.rename(columns={"Question":"content","Answer":"answer"})
|
26 |
df.drop('link to source (to prevent duplicate sources)',axis=1, inplace=True)
|
27 |
|