Subha95 commited on
Commit
f0b7bc3
·
verified ·
1 Parent(s): 3bcc980

Update chatbot_rag.py

Browse files
Files changed (1) hide show
  1. chatbot_rag.py +1 -1
chatbot_rag.py CHANGED
@@ -1,7 +1,7 @@
1
  from langchain_community.vectorstores import Chroma
2
  from langchain_community.embeddings import HuggingFaceEmbeddings
3
  from langchain_community.llms import HuggingFacePipeline
4
- from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
5
  from langchain.prompts import PromptTemplate
6
  from langchain_core.runnables import RunnablePassthrough
7
  from langchain_core.output_parsers import StrOutputParser
 
1
  from langchain_community.vectorstores import Chroma
2
  from langchain_community.embeddings import HuggingFaceEmbeddings
3
  from langchain_community.llms import HuggingFacePipeline
4
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, pipeline
5
  from langchain.prompts import PromptTemplate
6
  from langchain_core.runnables import RunnablePassthrough
7
  from langchain_core.output_parsers import StrOutputParser