Make prompt fully compliant with spec

cc

@ysharma

,

@ArthurZ

,

@osanseviero

app.py

@@ -34,12 +34,17 @@ examples=[
 
 
 def predict(message, chatbot):
-    
-    input_prompt = f"[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n "
+    # The first user message is _not_ stripped
+    do_strip = False
+    input_prompt = f"<s>[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n "
     for interaction in chatbot:
-        input_prompt = input_prompt + str(interaction[0]) + " [/INST] " + str(interaction[1]) + " </s><s> [INST] "
+        user_message = str(interaction[0])
+        user_message = user_message.strip() if do_strip else user_message
+        do_strip = True
+        input_prompt = input_prompt + user_message + " [/INST] " + str(interaction[1]).strip() + " </s><s>[INST] "
 
-    input_prompt = input_prompt + str(message) + " [/INST] "
+    message = str(message).strip() if do_strip else str(message)
+    input_prompt = input_prompt + message + " [/INST]"
 
     data = {
         "inputs": input_prompt,