Namitg02 commited on
Commit
b87b21e
·
verified ·
1 Parent(s): f4b0e4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -73,13 +73,16 @@ READER_MODEL="HuggingFaceH4/zephyr-7b-beta"
73
 
74
  #retrieval_qa_chat_prompt = hub.pull("langchain-ai/retrieval-qa-chat")
75
 
76
- qa_chat_prompt = ChatPromptTemplate.from_messages("""Answer the following question based only on the provided context:
77
-
78
- <context>
79
- {context}
80
- </context>
81
-
82
- Question: {input}""")
 
 
 
83
 
84
  docs_chain = create_stuff_documents_chain(
85
  "READER_MODEL", qa_chat_prompt
 
73
 
74
  #retrieval_qa_chat_prompt = hub.pull("langchain-ai/retrieval-qa-chat")
75
 
76
+ qa_chat_prompt = ChatPromptTemplate.from_messages(
77
+ [
78
+ SystemMessage(
79
+ content=(
80
+ "You are a Diabetes eductaor that provide advice to patients."
81
+ )
82
+ ),
83
+ HumanMessagePromptTemplate.from_template("{input}"),
84
+ ]
85
+ )
86
 
87
  docs_chain = create_stuff_documents_chain(
88
  "READER_MODEL", qa_chat_prompt