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

Update app.py

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