Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -481,21 +481,22 @@ def process_answer(question, full_text):
|
|
481 |
import tempfile
|
482 |
import os
|
483 |
import shutil
|
|
|
484 |
|
485 |
-
#
|
486 |
chroma_dir = os.path.join(tempfile.gettempdir(), "chroma_db")
|
487 |
|
488 |
-
# Clean up old
|
489 |
if os.path.exists(chroma_dir):
|
490 |
shutil.rmtree(chroma_dir)
|
491 |
os.makedirs(chroma_dir, exist_ok=True)
|
492 |
|
493 |
-
# Create
|
494 |
db = Chroma.from_documents(splits, embeddings, persist_directory=chroma_dir)
|
495 |
-
db.persist() # Optional: Saves data to disk for reuse within the same session
|
496 |
retriever = db.as_retriever()
|
497 |
|
498 |
|
|
|
499 |
# Set up the model
|
500 |
llm = load_model()
|
501 |
|
|
|
481 |
import tempfile
|
482 |
import os
|
483 |
import shutil
|
484 |
+
from langchain_community.vectorstores import Chroma
|
485 |
|
486 |
+
# Writable temp directory for Hugging Face Spaces or similar
|
487 |
chroma_dir = os.path.join(tempfile.gettempdir(), "chroma_db")
|
488 |
|
489 |
+
# Clean up old vectorstore
|
490 |
if os.path.exists(chroma_dir):
|
491 |
shutil.rmtree(chroma_dir)
|
492 |
os.makedirs(chroma_dir, exist_ok=True)
|
493 |
|
494 |
+
# Create Chroma vector store
|
495 |
db = Chroma.from_documents(splits, embeddings, persist_directory=chroma_dir)
|
|
|
496 |
retriever = db.as_retriever()
|
497 |
|
498 |
|
499 |
+
|
500 |
# Set up the model
|
501 |
llm = load_model()
|
502 |
|