Update app.py

app.py

@@ -76,13 +76,13 @@ class MistralChatBot:
         editorial_text =  re.sub("\n", " ¶ ", user_message)
     
         # Tokenize the prompt and check if it exceeds 500 tokens
-        num_tokens = len(tokenizer.tokenize(prompt))
+        num_tokens = len(tokenizer.tokenize(editorial_text))
     
         if num_tokens > 500:
             # Split the prompt into chunks
-            batch_prompts = split_text(prompt, max_tokens=500)
+            batch_prompts = split_text(editorial_text, max_tokens=500)
         else:
-            batch_prompts = [prompt]
+            batch_prompts = [editorial_text]
 
         out = token_classifier(batch_prompts)
         out = "".join(out)