Spaces:
Sleeping
Sleeping
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 |
)
|