1MR commited on
Commit
e5e6ae5
·
verified ·
1 Parent(s): 9528fa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,7 +83,7 @@ def get_conversation_chain(vectorstore, tokenH):
83
  def generate_response(messages):
84
  try:
85
  completion = client.chat.completions.create(
86
- model="mistralai/Mistral-7B-Instruct-v0.3",
87
  messages=messages,
88
  max_tokens=500
89
  )
@@ -97,7 +97,7 @@ def get_conversation_chain(vectorstore, tokenH):
97
  retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 5})
98
  documents = retriever.get_relevant_documents(user_input)
99
  documents_text = "\n".join(doc.page_content for doc in documents)
100
- messages = [{"role": "user", "content": user_input}, {"role": "assistant", "content": documents_text}]
101
  return generate_response(messages)
102
 
103
  return conversation_chain
 
83
  def generate_response(messages):
84
  try:
85
  completion = client.chat.completions.create(
86
+ model="Qwen/Qwen2.5-72B-Instruct",
87
  messages=messages,
88
  max_tokens=500
89
  )
 
97
  retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 5})
98
  documents = retriever.get_relevant_documents(user_input)
99
  documents_text = "\n".join(doc.page_content for doc in documents)
100
+ messages = [{"role": "user", "content": user_input}, {"role": "system", "content": documents_text}]
101
  return generate_response(messages)
102
 
103
  return conversation_chain