Update app.py

app.py

@@ -53,7 +53,7 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
         repetition_penalty=repetition_penalty,
     )
     for tokens in prediction:
-        tokens = re.findall(r'\s*\S+\s*', tokens)
+        tokens = re.findall(r'(.*?)(\s|$)', tokens)
         for s in tokens:
             answer = s
             history[-1][1] += answer