Update app.py

app.py

@@ -40,7 +40,7 @@ def run(message, chat_history, system_prompt, max_new_tokens=1024, temperature=0
     stream = client.generate_stream(prompt, **generate_kwargs)
     output = ''
     for response in stream:
-        if any([end_token in response.token_text for end_token in [EOS_STRING, EOT_STRING]]):
+        if any([end_token in response.token.text for end_token in [EOS_STRING, EOT_STRING]]):
             return output
         else:
             output += response.token.text