Update app.py

app.py

@@ -34,7 +34,7 @@ def generate(message, chat_history, max_new_tokens):
   final_prompt += "User: " + message + "\n"
   final_prompt += "Output:"
 
-  if len(tokenizer.tokenize(final_prompt)) >= 512 - max_new_tokens:
+  if len(tokenizer.tokenize(final_prompt)) >= 512:
     final_prompt = "Instruction: Say 'Input exceeded context size, please clear the chat history and retry!' Output:"
     
   # Streamer