Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,8 @@ retriever = vectordb.as_retriever(
|
|
57 |
)
|
58 |
|
59 |
|
60 |
-
from transformers import
|
|
|
61 |
|
62 |
llm_model = "HuggingFaceH4/zephyr-7b-beta"
|
63 |
pipe = pipeline(task="text-generation",llm_model = "models/HuggingFaceH4/zephyr-7b-beta",retriever = retriever)
|
|
|
57 |
)
|
58 |
|
59 |
|
60 |
+
from transformers import pipeline
|
61 |
+
from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
|
62 |
|
63 |
llm_model = "HuggingFaceH4/zephyr-7b-beta"
|
64 |
pipe = pipeline(task="text-generation",llm_model = "models/HuggingFaceH4/zephyr-7b-beta",retriever = retriever)
|