Yadukrishnan commited on
Commit
6a5b6b1
·
verified ·
1 Parent(s): b6bb995

Update src/llm_chain.py

Browse files
Files changed (1) hide show
  1. src/llm_chain.py +1 -1
src/llm_chain.py CHANGED
@@ -2,7 +2,7 @@ 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
6
 
7
  def setup_llm_chain():
8
  model, tokenizer = load_model()
 
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
6
 
7
  def setup_llm_chain():
8
  model, tokenizer = load_model()