Update src/demo.py

src/demo.py

@@ -40,6 +40,9 @@ def generate(input_text, sys_prompt) -> str:
 
     result = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
 
+    result = result.split(' [/INST]')
+    result = result[0] + '\n\n' + result[1]
+
     return result