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()
|