pradeepsengarr commited on
Commit
50a3fdd
·
verified ·
1 Parent(s): ed00ea1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -13
app.py CHANGED
@@ -478,21 +478,10 @@ def process_answer(question, full_text):
478
  embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
479
 
480
 
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
 
 
478
  embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
479
 
480
 
 
 
 
481
  from langchain_community.vectorstores import Chroma
482
 
483
+ # Create Chroma in-memory vector store
484
+ db = Chroma.from_documents(splits, embeddings)
 
 
 
 
 
 
 
 
485
  retriever = db.as_retriever()
486
 
487