jarif commited on
Commit
90c47ef
·
verified ·
1 Parent(s): 8cbef1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,6 +42,7 @@ def load_chroma_db():
42
  try:
43
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
44
  db = Chroma.load_local(chroma_dir, embeddings)
 
45
  return db.as_retriever()
46
  except Exception as e:
47
  st.error(f"Failed to load Chroma database: {e}")
@@ -90,4 +91,4 @@ def main():
90
  logger.exception("Unexpected error in main function")
91
 
92
  if __name__ == '__main__':
93
- main()
 
42
  try:
43
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
44
  db = Chroma.load_local(chroma_dir, embeddings)
45
+ logger.info(f"Chroma database loaded successfully from {chroma_dir}")
46
  return db.as_retriever()
47
  except Exception as e:
48
  st.error(f"Failed to load Chroma database: {e}")
 
91
  logger.exception("Unexpected error in main function")
92
 
93
  if __name__ == '__main__':
94
+ main()