Spaces:
Sleeping
Sleeping
Logeswaransr
commited on
Commit
•
afb5ebd
1
Parent(s):
8a2282e
Update initialize.py
Browse files- 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 |
-
|
|
|
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':
|
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 |
)
|