Update app.py
Browse files
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()
|