Yadukrishnan commited on
Commit
020d75a
·
verified ·
1 Parent(s): 50c4016

Update src/llm_chain.py

Browse files
Files changed (1) hide show
  1. src/llm_chain.py +1 -1
src/llm_chain.py CHANGED
@@ -1,5 +1,5 @@
1
  from transformers import pipeline
2
- from langchain_community.llms import HuggingFacePipeline
3
  from langchain.prompts import PromptTemplate
4
  from langchain_core.output_parsers import StrOutputParser
5
  from .model_loader import load_model
 
1
  from transformers import pipeline
2
+ from langchain_huggingface import HuggingFacePipeline
3
  from langchain.prompts import PromptTemplate
4
  from langchain_core.output_parsers import StrOutputParser
5
  from .model_loader import load_model