Hugging Face's logo

Spaces:
ale979
/
Smart-Assistant
Build error

App Files Community
ale979 commited on
Commit
714e8f2
·
verified ·
1 Parent(s): bd60ba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ embedding_function = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
27
  def answer_query(message, chat_history):
28
  base_compressor = LLMChainExtractor.from_llm(chat)
29
 
30
- db = Chroma(persist_directory = "./slide_chromaDB4", embedding_function=embedding_function)
31
  base_retriever = db.as_retriever()
32
  mq_retriever = MultiQueryRetriever.from_llm(retriever = base_retriever, llm=chat)
33
  compression_retriever = ContextualCompressionRetriever(base_compressor=base_compressor, base_retriever=mq_retriever)
 
27
  def answer_query(message, chat_history):
28
  base_compressor = LLMChainExtractor.from_llm(chat)
29
 
30
+ db = Chroma(persist_directory = "./slide_chromaDB4", embedding_function=embedding_function, collection_name="slideCollection")
31
  base_retriever = db.as_retriever()
32
  mq_retriever = MultiQueryRetriever.from_llm(retriever = base_retriever, llm=chat)
33
  compression_retriever = ContextualCompressionRetriever(base_compressor=base_compressor, base_retriever=mq_retriever)