Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,18 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|
|
60 |
def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
|
61 |
print(f'message is - {message}')
|
62 |
print(f'history is - {history}')
|
63 |
-
conversation = [{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
64 |
for prompt, answer in history:
|
65 |
conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
|
66 |
conversation.append({"role": "user", "content": message})
|
|
|
60 |
def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
|
61 |
print(f'message is - {message}')
|
62 |
print(f'history is - {history}')
|
63 |
+
conversation = [{
|
64 |
+
"role": "system",
|
65 |
+
"content": (
|
66 |
+
"Anda adalah chatbot kesehatan masyarakat yang hanya memberikan informasi dan konsultasi terkait stunting. "
|
67 |
+
"Jawab hanya pertanyaan yang berhubungan dengan pencegahan stunting, penyebab stunting, dampak stunting, "
|
68 |
+
"intervensi gizi, perilaku hidup bersih dan sehat (PHBS), serta kebijakan atau program pemerintah terkait stunting. "
|
69 |
+
"Jika pengguna mengajukan pertanyaan di luar topik stunting, tolak dengan sopan dan tegas dengan mengatakan: "
|
70 |
+
"\"Maaf, saya hanya bisa membantu terkait topik stunting. Untuk pertanyaan di luar itu, silakan konsultasikan dengan pihak atau layanan yang sesuai.\" "
|
71 |
+
"Jangan menjawab atau berspekulasi terhadap pertanyaan yang tidak relevan dengan stunting. Fokuskan seluruh respons hanya pada konteks edukasi, pencegahan, dan penanggulangan stunting."
|
72 |
+
)
|
73 |
+
}]
|
74 |
+
|
75 |
for prompt, answer in history:
|
76 |
conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
|
77 |
conversation.append({"role": "user", "content": message})
|