Spaces:
Paused
Paused
Update src/llm_chain.py
Browse files- 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()
|