Namitg02 commited on
Commit
4adfdbc
·
verified ·
1 Parent(s): 15ac750

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -59,10 +59,6 @@ retriever = vectordb.as_retriever(
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)
65
-
66
  from langchain_core.messages import SystemMessage
67
  from langchain_core.prompts import HumanMessagePromptTemplate
68
  from langchain_core.prompts import ChatPromptTemplate
@@ -78,7 +74,11 @@ qa_chat_prompt = ChatPromptTemplate.from_messages(
78
  ]
79
  )
80
 
81
- chain = qa_chat_prompt | pipe
 
 
 
 
82
 
83
  import gradio as gr
84
  #ragdemo = gr.load("models/HuggingFaceH4/zephyr-7b-beta")
 
59
 
60
  from transformers import pipeline
61
  from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
 
 
 
 
62
  from langchain_core.messages import SystemMessage
63
  from langchain_core.prompts import HumanMessagePromptTemplate
64
  from langchain_core.prompts import ChatPromptTemplate
 
74
  ]
75
  )
76
 
77
+ llm_model = "HuggingFaceH4/zephyr-7b-beta"
78
+ pipe = pipeline(task="text-generation",llm_model = "models/HuggingFaceH4/zephyr-7b-beta",retriever = retriever,chat = qa_chat_prompt)
79
+
80
+
81
+ #chain = qa_chat_prompt | pipe
82
 
83
  import gradio as gr
84
  #ragdemo = gr.load("models/HuggingFaceH4/zephyr-7b-beta")