chahah commited on
Commit
3c25581
·
verified ·
1 Parent(s): 001364c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -39,7 +39,8 @@ embeddings = HuggingFaceInstructEmbeddings(model_name=embed_model)
39
 
40
  def initialize(arxivcode):
41
  docs = retriever.invoke(str(arxivcode))
42
- docs[0].metadata['Published'] = str(docs[0].metadata['Published'])
 
43
 
44
  def format_docs(docs):
45
  return "\n\n".join(doc.page_content for doc in docs)
 
39
 
40
  def initialize(arxivcode):
41
  docs = retriever.invoke(str(arxivcode))
42
+ for i in range(len(docs)):
43
+ docs[i].metadata['Published'] = str(docs[i].metadata['Published'])
44
 
45
  def format_docs(docs):
46
  return "\n\n".join(doc.page_content for doc in docs)