Update app.py

app.py

@@ -71,17 +71,12 @@ def predict(message, history, system_prompt, temperature, max_new_tokens, top_k,
         instruction += ' ' + message + ' [/INST]'
     elif CHAT_TEMPLATE == "Bielik":
         stop_tokens = ["</s>"]
-        prompt_builder = ["<s>"]
-        if history:
-            for human, assistant in history:
-                if system_prompt:
-                    prompt_builder.append(f"[INST] <<SYS>>\n{system_prompt}\n<</SYS>>\n\n{human} [/INST] {assistant}</s>")
-                    system_prompt = None
-                else:
-                    prompt_builder.append(f"[INST] {human} [/INST] {assistant}</s>")
-            prompt_builder.append(f"[INST] {message} [/INST]")
-        else:
-            prompt_builder.append(f"[INST] <<SYS>>\n{system_prompt}\n<</SYS>>\n\n{message} [/INST]")
+        prompt_builder = ["<s>[INST] "]
+        if system_prompt:
+            prompt_builder.append(f"<<SYS>>\n{system_prompt}\n<</SYS>>\n\n")
+        for human, assistant in history:
+            prompt_builder.append(f"{human} [/INST] {assistant}</s>[INST] "))
+        prompt_builder.append(f"{message} [/INST]")
         instruction = ''.join(prompt_builder)
     else:
         raise Exception("Incorrect chat template, select 'ChatML' or 'Mistral Instruct'")