simple prompt

app.py

@@ -60,10 +60,11 @@ def detect(attacked_text, window_size, method, prompt):
     return label
 
 def get_prompt(message: str) -> str:
-    texts = [f'<s>[INST] <<SYS>>\n{DEFAULT_SYSTEM_PROMPT}\n<</SYS>>\n\n']
+    # texts = [f'<s>[INST] <<SYS>>\n{DEFAULT_SYSTEM_PROMPT}\n<</SYS>>\n\n']
     # The first user input is _not_ stripped
-    texts.append(f'{message} [/INST]')
-    return ''.join(texts)
+    # texts.append(f'{message} [/INST]')
+    # return ''.join(texts)
+    return f"[INST] {message} [/INST]"
 
 
 with gr.Blocks() as demo: