tstone87 commited on
Commit
d073b51
·
verified ·
1 Parent(s): 2d626a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,13 +51,13 @@ def get_huggingface_response(query):
51
  messages = [{"role": "system", "content": "Provide accurate food stamp information for Colorado."},
52
  {"role": "user", "content": query}]
53
  response = ""
54
- for message in client.chat_completion(messages, max_tokens=512, stream=True, temperature=0.7, top_p=0.95):
55
  response += message.choices[0].delta.content
56
  return response
57
 
58
  # 🔹 Main Chatbot Function
59
  def chatbot_response(message, history):
60
- relevant_info = search_faiss(message, top_k=3) # Retrieve 3 most relevant sections
61
 
62
  if "No relevant information found." not in relevant_info:
63
  user_query_with_context = f"""
 
51
  messages = [{"role": "system", "content": "Provide accurate food stamp information for Colorado."},
52
  {"role": "user", "content": query}]
53
  response = ""
54
+ for message in client.chat_completion(messages, max_tokens=1024, stream=True, temperature=0.7, top_p=0.95):
55
  response += message.choices[0].delta.content
56
  return response
57
 
58
  # 🔹 Main Chatbot Function
59
  def chatbot_response(message, history):
60
+ relevant_info = search_faiss(message, top_k=1) # Retrieve 1 most relevant sections
61
 
62
  if "No relevant information found." not in relevant_info:
63
  user_query_with_context = f"""