Update main.py

main.py

@@ -54,7 +54,7 @@ def generate(prompt, history, temperature=0.9, max_new_tokens=512, top_p=0.95, r
             word += decoded_text
 
             # If the token is a space or the end of the stream, add the word to the output and reset the word
-            if token_text == " " or response.is_end_of_sequence:
+            if token_text == " " or response.is_end_of_stream:
                 output += word + " "
                 word = ""