kviraj722 commited on
Commit
3500949
·
verified ·
1 Parent(s): 8e84a2d

Update rag_pipeline.py

Browse files
Files changed (1) hide show
  1. rag_pipeline.py +1 -1
rag_pipeline.py CHANGED
@@ -12,7 +12,7 @@ load_dotenv()
12
  client = Groq(api_key=os.environ["GROQ_API_KEY"])
13
  CHROMA_DIR = "./chroma_db"
14
  # Embedding model (free HuggingFace)
15
- embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
16
 
17
  def process_file(file_bytes, filename, file_id):
18
  ext = filename.split('.')[-1].lower()
 
12
  client = Groq(api_key=os.environ["GROQ_API_KEY"])
13
  CHROMA_DIR = "./chroma_db"
14
  # Embedding model (free HuggingFace)
15
+ embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={"trust_remote_code": True})
16
 
17
  def process_file(file_bytes, filename, file_id):
18
  ext = filename.split('.')[-1].lower()