assistant word deleted

app.py

@@ -41,8 +41,7 @@ def predict(message, history):
 
     partial_message = ""
     for new_token in streamer:
-        print(new_token)
-        if new_token != 'assistant':
+        if  'assistant' not in new_token:
             partial_message += new_token
             yield partial_message