Spaces:
Runtime error
Runtime error
Update rag_pipeline.py
Browse files- 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()
|