Penality commited on
Commit
b0919a4
·
verified ·
1 Parent(s): 44817d0

Update app.py

Browse files

return data on store_document_data to a dict

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def store_document_data(PDF_FILE, METADATA_FILE, INDEX_FILE):
67
  metadata[str(doc_index)] = PDF_FILE
68
  print(" Saved Metadata")
69
 
70
- return {"metadata": metadata, "index": index}
71
 
72
  def retrieve_document(query):
73
  print(f"Retrieving document based on:\n{query}")
 
67
  metadata[str(doc_index)] = PDF_FILE
68
  print(" Saved Metadata")
69
 
70
+ return [{"metadata": metadata}, {"index": index}]
71
 
72
  def retrieve_document(query):
73
  print(f"Retrieving document based on:\n{query}")