Spaces:
Sleeping
Sleeping
Update chatbot_rag.py
Browse files- chatbot_rag.py +2 -1
chatbot_rag.py
CHANGED
|
@@ -6,7 +6,8 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
|
|
| 6 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
| 7 |
from langchain.llms import HuggingFacePipeline
|
| 8 |
from langchain.chains import RetrievalQA
|
| 9 |
-
|
|
|
|
| 10 |
|
| 11 |
def build_qa():
|
| 12 |
"""Builds and returns the RAG QA pipeline."""
|
|
|
|
| 6 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
| 7 |
from langchain.llms import HuggingFacePipeline
|
| 8 |
from langchain.chains import RetrievalQA
|
| 9 |
+
from langchain_huggingface import HuggingFacePipeline, HuggingFaceEmbeddings
|
| 10 |
+
from langchain_chroma import Chroma
|
| 11 |
|
| 12 |
def build_qa():
|
| 13 |
"""Builds and returns the RAG QA pipeline."""
|