Update RAG.py

RAG.py

@@ -49,9 +49,19 @@ def load_models_llm():
 #     return vector_store
 
 def create_database(embedding, documents):
-    vector_store = Chroma.from_documents(documents, embedding=embedding)
+    # Use a local directory for Chroma database storage
+    vector_store = Chroma.from_documents(
+        documents,
+        embedding=embedding,
+        persist_directory="/chroma_db"  # Hugging Face Spaces allows usage of this directory
+    )
     return vector_store
 
+
+# def create_database(embedding, documents):
+#     vector_store = Chroma.from_documents(documents, embedding=embedding)
+#     return vector_store
+
 # retriever = create_database().as_retriever()