Namitg02 commited on
Commit
e69e635
·
verified ·
1 Parent(s): 9f27381

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -57,7 +57,8 @@ retriever = vectordb.as_retriever(
57
  )
58
 
59
 
60
- from transformers import huggingfacepipeline, pipeline
 
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)