Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|
|
44 |
def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
|
45 |
print(f'message is - {message}')
|
46 |
print(f'history is - {history}')
|
47 |
-
conversation = [{"role": "system", "content": '
|
48 |
for prompt, answer in history:
|
49 |
conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
|
50 |
conversation.append({"role": "user", "content": message})
|
|
|
44 |
def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
|
45 |
print(f'message is - {message}')
|
46 |
print(f'history is - {history}')
|
47 |
+
conversation = [{"role": "system", "content": 'Anda adalah chatbot kesehatan masyarakat yang hanya memberikan informasi dan konsultasi terkait pencegahan stunting, gizi anak, dan kesehatan ibu. Tolak semua pertanyaan yang tidak relevan atau di luar konteks ini dengan sopan.'}]
|
48 |
for prompt, answer in history:
|
49 |
conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
|
50 |
conversation.append({"role": "user", "content": message})
|