pradeepsengarr commited on
Commit
ed00ea1
·
verified ·
1 Parent(s): 2183bfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -481,21 +481,22 @@ def process_answer(question, full_text):
481
  import tempfile
482
  import os
483
  import shutil
 
484
 
485
- # Use a safe, writable temp directory
486
  chroma_dir = os.path.join(tempfile.gettempdir(), "chroma_db")
487
 
488
- # Clean up old data and recreate
489
  if os.path.exists(chroma_dir):
490
  shutil.rmtree(chroma_dir)
491
  os.makedirs(chroma_dir, exist_ok=True)
492
 
493
- # Create vectorstore and retriever
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