Subha95 commited on
Commit
1933718
·
verified ·
1 Parent(s): edf6046

Update chatbot_rag.py

Browse files
Files changed (1) hide show
  1. 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."""