Logeswaransr commited on
Commit
afb5ebd
1 Parent(s): 8a2282e

Update initialize.py

Browse files
Files changed (1) hide show
  1. initialize.py +3 -2
initialize.py CHANGED
@@ -21,7 +21,8 @@ def define_components(document_store, api_key):
21
  retriever = InMemoryBM25Retriever(document_store, top_k=3)
22
 
23
  template = """
24
- Given the following information, answer the question.
 
25
 
26
  Context:
27
  {% for document in documents %}
@@ -37,7 +38,7 @@ def define_components(document_store, api_key):
37
  model="mistralai/Mistral-7B-Instruct-v0.1",
38
  token=Secret.from_token(api_key),
39
  generation_kwargs = {
40
- 'max_new_tokens':100,
41
  'temperature':0.7
42
  }
43
  )
 
21
  retriever = InMemoryBM25Retriever(document_store, top_k=3)
22
 
23
  template = """
24
+ You are a Chatbot designed to spread Awareness about Alzheimer's Disease. You are AI Chaperone.
25
+ You will be provided information about Alzheimer's Disease as context for each question. Given the following information, answer the question.
26
 
27
  Context:
28
  {% for document in documents %}
 
38
  model="mistralai/Mistral-7B-Instruct-v0.1",
39
  token=Secret.from_token(api_key),
40
  generation_kwargs = {
41
+ 'max_new_tokens':50,
42
  'temperature':0.7
43
  }
44
  )