Spaces:
Runtime error
Runtime error
bugfix: Fix embed_queries() KeyError
Browse files
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 |
|